adding the unit test for feature extraction