def test_handle_conditions_not_exist(self): conditions: dict = {'NM': 2, 'D': 0} new_condition = 'RE' updated_conditions = DataImport.handle_conditions( conditions, new_condition) self.assertEqual(updated_conditions, conditions)
def test_handle_conditions_add(self): conditions: dict = {'NM': 2, 'D': 0} new_condition = 'D' updated_conditions = DataImport.handle_conditions( conditions, new_condition) self.assertEqual(updated_conditions, {'NM': 2, 'D': 1})