def test_b(self):
"""Test no intake are available for level control.
There is an intake available for level control but that belongs to
another open water.
"""
open_water = OpenWater()
open_water.pk = 2
intake = find_pumping_station_level_control(open_water, True)
self.assertTrue(intake is None)
def test_e(self):
"""Test no intakes are available for level control.
There is a pump available for level control.
"""
self.pumping_station.into = False
self.pumping_station.save()
open_water = self.pumping_station.open_water
intake = find_pumping_station_level_control(open_water, True)
self.assertTrue(intake is None)
def test_d(self):
"""Test no intakes are available for level control.
There is an intake available but that intake cannot be used for level
control.
"""
self.pumping_station.computed_level_control = False
self.pumping_station.save()
open_water = self.pumping_station.open_water
intake = find_pumping_station_level_control(open_water, True)
self.assertTrue(intake is None)
def test_c(self):
"""Test multiple intakes are available for level control."""
pumping_station = self.create_pumping_station()
self.assertEqual(2, PumpingStation.objects.count())
open_water = pumping_station.open_water
intake = find_pumping_station_level_control(open_water, True)
self.assertTrue((intake == self.pumping_station) ^
(intake == pumping_station))
pumping_station.delete()
def test_a(self):
"""Test one intake is available for level control."""
open_water = self.pumping_station.open_water
intake = find_pumping_station_level_control(open_water, True)
self.assertEqual(self.pumping_station, intake)
def test_no_intakes_or_pumps_exist():
"""Test no intakes or pumps exist."""
open_water = OpenWater()
open_water.pk = 1
intake = find_pumping_station_level_control(open_water, True)
assert intake is None