def setUp(self):
self.data = {
100: toggle_doors(generate_doors(100)),
999: toggle_doors(generate_doors(999))
}
def test_collect_open_doors_10_toggled(self):
result = collect_open_doors(toggle_doors(generate_doors(10)))
self.assertEqual(result, ["1", "4", "9"])