def testGetNeighborZones_noNeighborTypesSpecified_returnsCorrectList(self): zone1 = Zone('foyer') zone2 = Zone('porch') zone3 = Zone('office') zone1 = zone1.addNeighbor( Neighbor(zone2.getId(), NeighborType.OPEN_SPACE)) zone1 = zone1.addNeighbor( Neighbor(zone3.getId(), NeighborType.OPEN_SPACE_MASTER)) zm = MockedZoneManager([zone1, zone2, zone3]) self.assertEqual(2, len(zone1.getNeighborZones(zm)))
def testGetNeighborZones_noZoneManager_throwsException(self): zone1 = Zone('ff') with self.assertRaises(ValueError) as cm: zone1.getNeighborZones(None) self.assertEqual('zoneManager must not be None', cm.exception.args[0])