From adf0261bd29e818dd925b0e5b8dc7979c35e3b71 Mon Sep 17 00:00:00 2001 From: "Christian T. Jacobs" Date: Thu, 6 Jul 2017 22:17:15 +0100 Subject: [PATCH] More tests for prefix/suffix removal. --- tests/test_callsign_lookup.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) 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. """