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"])