def test_query_time_interval(self):
d = parking.ParkingRateDataStore()
rates = [
parking.ParkingRate(3, (0, 1), (0, 2), 0),
parking.ParkingRate(3, (0, 2), (0, 5), 1),
parking.ParkingRate(3, (0, 5), (0, 8), 2),
]
d.add_parking_rates(rates)
r = d.query_time_interval((0, 3), (0, 4), 3)
self.assertEqual(r, 1)
def test_query_time_interval_no_result(self):
d = parking.ParkingRateDataStore()
rates = [
parking.ParkingRate(3, (0, 1), (0, 2), 0),
parking.ParkingRate(3, (0, 2), (0, 5), 1),
parking.ParkingRate(3, (0, 8), (0, 9), 2),
]
d.add_parking_rates(rates)
r = d.query_time_interval((0, 6), (0, 7), 3)
self.assertIsNone(r)
def test_add_parking_rate(self):
d = parking.ParkingRateDataStore()
for wd in range(0, 6):
r = parking.ParkingRate(wd, 1, 2, wd)
d.add_parking_rate(r)
i = d.weekday_interval_trees[wd].all_intervals.pop()
self.assertEqual(i.data, wd)
def test_add_parking_rate(self):
d = parking.ParkingRateDataStore()
rates = [parking.ParkingRate(wd, 1, 2, wd) for wd in range(0, 6)]
d.add_parking_rates(rates)
for r in rates:
i = d.weekday_interval_trees[r.weekday].all_intervals.pop()
self.assertEqual(i.data, r.weekday)
def test_add_parking_rate(self):
r = parking.ParkingRate(3, 1, 2, 9000)
self.assertEqual(r.weekday, 3)
self.assertEqual(r.begin, 1)
self.assertEqual(r.end, 2)
self.assertEqual(r.rate, 9000)