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)
def test_box_source3(self):
box = BoxSource('SOURCE')
box.add(100)
self.assertEqual(box.input(), 100)
box.step()
self.assertEqual(box.output(), 100)
box.remove(10)
self.assertEqual(box.removed(), 10)
box.step()
self.assertEqual(box.removed(1), 10)
self.assertEqual(box.removed(), 0)
self.assertEqual(box.output(), 90)
box.remove(10)
box.remove(10)
self.assertEqual(box.removed(), 20)
box.step()
self.assertEqual(box.removed(1), 20)
self.assertEqual(box.removed(), 0)
self.assertEqual(box.output(), 70)