Add an unit test for the expansion