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)