def test_get_total_points_more_belts(): more_belts = dict(brown=BeltStats(400, 2), black=BeltStats(600, 5)) # this way to dict merge is >= 3.5 (PEP 448) ninja_belts_updated = {**ninja_belts, **more_belts} assert get_total_points(ninja_belts_updated) == 6475
def test_get_total_points_given_belts(): assert get_total_points() == 2675