diff --git a/tests/test_callsign_lookup.py b/tests/test_callsign_lookup.py index 8c5aff4..37747e3 100644 --- a/tests/test_callsign_lookup.py +++ b/tests/test_callsign_lookup.py @@ -46,13 +46,17 @@ class TestCallsignLookup(unittest.TestCase): def test_strip_prefix_only(self): """ Check that a callsign with only a prefix is stripped correctly. """ - callsign = "F/MYCALL" - assert strip(callsign) == "MYCALL" + assert strip("F/MYCALL") == "MYCALL" def test_strip_suffix_only(self): """ Check that a callsign with only a suffix is stripped correctly. """ - callsign = "MYCALL/M" - assert strip(callsign) == "MYCALL" + assert strip("MYCALL/M") == "MYCALL" + assert strip("MYCALL/P") == "MYCALL" + assert strip("MYCALL/A") == "MYCALL" + assert strip("MYCALL/MM") == "MYCALL" + assert strip("MYCALL/PM") == "MYCALL" + assert strip("MYCALL/AM") == "MYCALL" + assert strip("MYCALL/QRP") == "MYCALL" def test_strip_no_prefix_or_suffix(self): """ Check that a callsign with no prefix or suffix remains unmodified. """