def test_box_source4(self):
box = BoxSource('SOURCE')
box.add(100)
box.step()
self.assertEqual(box.size(), 100)
self.assertEqual(box.output(), 100)
box.step()
self.assertEqual(box.size(), 100)
self.assertEqual(box.output(), 100)
box.remove(10)
box.step()
self.assertEqual(box.size(), 90)
self.assertEqual(box.output(), 90)
def test_box_source2(self):
box = BoxSource('SOURCE')
box.add(100)
self.assertEqual(box.input(), 100)
box.step()
self.assertEqual(box.output(), 100)
box.step()
self.assertEqual(box.output(), 100)
box.add(10)
box.step()
self.assertEqual(box.input(), 0)
self.assertEqual(box.size(), 110)
self.assertEqual(box.output(), 110)