def test_correct_params(self): valid_menus = [{'root_id': 2, 'children': [4, 6]}] invalid_menus = [{'root_id': 1, 'children': [1, 3, 5]}] result = menus.render_aggregate_menus_json(valid_menus, invalid_menus) expected_result = ( '{"valid_menus": [{"root_id": 2, "children": [4, 6]}], ' '"invalid_menus": [{"root_id": 1, "children": [1, 3, 5]}]}') self.assertEqual(expected_result, result)
import menus if __name__ == '__main__': problem_id = input("Please enter a problem set id: ") menus_info = menus.fetch_menus_data(problem_id) products, root_ids = menus.parse_menus_data(menus_info) valid_menus, invalid_menus = menus.validate_menus(root_ids, products) print(menus.render_aggregate_menus_json(valid_menus, invalid_menus))
def test_no_menus_supplied(self): result = menus.render_aggregate_menus_json() expected_result = '{"valid_menus": [], "invalid_menus": []}' self.assertEqual(expected_result, result)
def test_both_empty_menus(self): result = menus.render_aggregate_menus_json([], []) expected_result = '{"valid_menus": [], "invalid_menus": []}' self.assertEqual(expected_result, result)