from backend.sector import detect_sector def test_bank_text_detected(): text = ( "The bank reported total deposits and advances. NIM improved. " "Gross NPA declined. CASA ratio stood at 45%. RBI guidelines followed." ) assert detect_sector(text) == "BANK" def test_it_text_detected(): text = ( "TCS reported strong growth across IT services. " "Attrition was 12%. Offshore delivery and software exports grew." ) assert detect_sector(text) == "IT" def test_generic_text_falls_back_to_general(): assert detect_sector("The company reported strong results this year.") == "GENERAL" def test_fewer_than_three_keyword_hits_is_general(): # one keyword alone shouldn't flip the sector assert detect_sector("Some deposits were made.") == "GENERAL"