def test_case_2(self):
blocks = [
{
"gym": False,
"office": True,
"school": True,
"store": False,
},
{
"gym": True,
"office": False,
"school": False,
"store": False,
},
{
"gym": True,
"office": False,
"school": True,
"store": False,
},
{
"gym": False,
"office": False,
"school": True,
"store": False,
},
{
"gym": False,
"office": False,
"school": True,
"store": True,
},
]
reqs = ["gym", "office", "school", "store"]
self.assertEqual(program.apartmentHunting(blocks, reqs), 2)
def test_case_4(self):
blocks = [
{
"foo": True,
"gym": False,
"kappa": False,
"office": True,
"school": True,
"store": False,
},
{
"foo": True,
"gym": True,
"kappa": False,
"office": False,
"school": False,
"store": False,
},
{
"foo": True,
"gym": True,
"kappa": False,
"office": False,
"school": True,
"store": False,
},
{
"foo": True,
"gym": False,
"kappa": False,
"office": False,
"school": True,
"store": False,
},
{
"foo": True,
"gym": True,
"kappa": False,
"office": False,
"school": True,
"store": False,
},
{
"foo": True,
"gym": False,
"kappa": False,
"office": False,
"school": True,
"store": True,
},
]
reqs = ["gym", "school", "store"]
self.assertEqual(
program.apartmentHunting(blocks, reqs) in [4, 5], True)