Add in tests for feature generation tasks