marcel/pylot
3
0
Fork 0
Code Issues 17 Pull Requests 1 Releases Wiki Activity

[minor] improved unittest result

Browse Source
This commit is contained in:
Marcel Paffrath 2024-09-09 16:05:02 +02:00
parent 03947d2363
commit cc7716a2b7
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_autopickstation/test_autopickstation.py
View File

@ -272,7 +272,8 @@ class TestAutopickStation(unittest.TestCase):
with HidePrints(): with HidePrints():
result, station = autopickstation(wfstream=wfstream, pickparam=self.pickparam_taupy_disabled, result, station = autopickstation(wfstream=wfstream, pickparam=self.pickparam_taupy_disabled,
metadata=(None, None)) metadata=(None, None))
compare_dicts(result, expected) compare_dicts(result=result['P'], expected=expected['P'])
compare_dicts(result=result['S'], expected=expected['S'])
self.assertEqual('GRA1', station) self.assertEqual('GRA1', station)
def test_autopickstation_a106_taupy_enabled(self): def test_autopickstation_a106_taupy_enabled(self):
@ -290,7 +291,8 @@ class TestAutopickStation(unittest.TestCase):
with HidePrints(): with HidePrints():
result, station = autopickstation(wfstream=self.a106, pickparam=self.pickparam_taupy_enabled, result, station = autopickstation(wfstream=self.a106, pickparam=self.pickparam_taupy_enabled,
metadata=self.metadata, origin=self.origin) metadata=self.metadata, origin=self.origin)
compare_dicts(result=result, expected=expected) compare_dicts(result=result['P'], expected=expected['P'])
compare_dicts(result=result['S'], expected=expected['S'])
def test_autopickstation_station_missing_in_metadata(self): def test_autopickstation_station_missing_in_metadata(self):
@ -312,7 +314,8 @@ class TestAutopickStation(unittest.TestCase):
with HidePrints(): with HidePrints():
result, station = autopickstation(wfstream=self.a005a, pickparam=self.pickparam_taupy_enabled, result, station = autopickstation(wfstream=self.a005a, pickparam=self.pickparam_taupy_enabled,
metadata=self.metadata, origin=self.origin) metadata=self.metadata, origin=self.origin)
compare_dicts(result, expected) compare_dicts(result=result['P'], expected=expected['P'])
compare_dicts(result=result['S'], expected=expected['S'])
def run_dict_comparison(result, expected): def run_dict_comparison(result, expected):
@ -332,8 +335,8 @@ def compare_dicts(result, expected):
run_dict_comparison(result, expected) run_dict_comparison(result, expected)
except AssertionError: except AssertionError:
raise AssertionError(f'Dictionaries not equal.' raise AssertionError(f'Dictionaries not equal.'
f'\n<<Expected>>: \n\n{pretty_print_dict(expected)}' f'\n\n<<Expected>>\n{pretty_print_dict(expected)}'
f'\n<<Result>>: \n{pretty_print_dict(result)}') f'\n\n<<Result>>\n{pretty_print_dict(result)}')
def pretty_print_dict(dct): def pretty_print_dict(dct):