tests/test_intent_extraction.py

from src.intent_extractor import IntentExtractor

def test_extract():
    extractor = IntentExtractor()
    intents = extractor.extract(['book','flight'])
    assert 'book' in intents