Add tests for src/feature_extraction.py