class IntersectionPoolTests(unittest.TestCase):
def setUp(self):
self.sts_EW = ('A', 'B', 'C', 'D')
self.sts_NS = ('1', '2', '3')
self.intersectionpool = GridIntersectionPool(self.sts_NS, self.sts_EW)
def _random_intersection(self):
st1 = random.choice(self.sts_NS)
st2 = random.choice(self.sts_EW)
return Intersection(st1, st2)
# TODO: add test for getting intersection that does not existZ
def test_get_one_intersection(self):
intersection_returned = self.intersectionpool.getat('B', '2')
intersection_expected = Intersection('B', '2')
self.assertEqual(intersection_expected, intersection_returned)
def setUp(self):
self.sts_EW = ('A', 'B', 'C', 'D')
self.sts_NS = ('1', '2', '3')
self.intersectionpool = GridIntersectionPool(self.sts_NS, self.sts_EW)
