class ReservoirWellTest(unittest.TestCase):
def setUp(self):
self.reservoir = Reservoir()
self.row = self.reservoir.row(1)
def liquid_allocation_test(self):
set_vol = 50
self.row.allocate(water=set_vol)
vol = self.row.get_volume()
self.assertEqual(vol, set_vol)
def liquid_capacity_test(self):
with self.assertRaises(x.LiquidOverflow):
self.row.allocate(water=10000, ml=True)
def liquid_total_capacity_test(self):
row = self.row
row.allocate(water=20, ml=True)
row.add_liquid(water=1, ml=True)
with self.assertRaises(x.LiquidOverflow):
row.add_liquid(water=1, ml=True)
def named_liquid_test(self):
self.row.add_named_liquid(100, 'water')
self.assertEqual(self.row.get_volume('water'), 100)
def liquid_total_mixture_test(self):
self.row.allocate(water=20000)
self.row.add_liquid(buffer=1000)
with self.assertRaises(x.LiquidOverflow):
self.row.add_liquid(saline=1)
class ReservoirTest(unittest.TestCase):
expected_margin = Reservoir.spacing
def setUp(self):
self.reservoir = Reservoir()
def row_sanity_test(self):
"""
Row sanity check.
"""
row = self.reservoir.rows + 1
with self.assertRaises(x.SlotMissing):
self.reservoir.get_child('A{}'.format(row))
def col_sanity_test(self):
"""
Column sanity check.
"""
with self.assertRaises(x.SlotMissing):
self.reservoir.get_child('B1')
class LiquidDebtTest(unittest.TestCase):
_allow_liquid_debt = None # Store library value to restore.
_allow_unspec = None # Store library value to restore.
def setUp(self):
self._allow_liquid_debt = LiquidInventory._allow_liquid_debt
self._allow_unspec = LiquidInventory._allow_unspecified_liquids
LiquidInventory._allow_liquid_debt = True
self.reservoir = Reservoir()
self.row = self.reservoir.row(1)
def tearDown(self):
LiquidInventory._allow_liquid_debt = self._allow_liquid_debt
LiquidInventory._allow_unspecified_liquids = self._allow_unspec
def test_liquid_total_capacity_test(self):
"""
Supports negative liquid capacity.
"""
row1 = self.reservoir.row(1)
row2 = self.reservoir.row(2)
row1.transfer(1000, row2, name="water")
self.assertEqual(row1.get_volume(), -1000)
self.assertEqual(row1.get_volume("water"), -1000)
def liquid_named_debt_test(self):
"""
Only allows for named liquid debt.
"""
LiquidInventory._allow_liquid_debt = True
LiquidInventory._allow_unspecified_liquids = False
row1 = self.reservoir.row(1)
row2 = self.reservoir.row(2)
with self.assertRaises(x.DataMissing):
# This shouldn't work because we have no idea what they're
# transfering, so it's not very useful to us.
row1.transfer(1000, row2)
def setUp(self):
self._allow_liquid_debt = LiquidInventory._allow_liquid_debt
self._allow_unspec = LiquidInventory._allow_unspecified_liquids
LiquidInventory._allow_liquid_debt = True
self.reservoir = Reservoir()
self.row = self.reservoir.row(1)
def setUp(self):
self.reservoir = Reservoir()
self.row = self.reservoir.row(1)
def setUp(self):
self.reservoir = Reservoir()
