def test_ifRemoveWorks(self):
bc = BallContainer()
bc.add(BallColors.BLACK, 5)
bc.add(BallColors.WHITE, 2)
bc.removeOne(BallColors.WHITE)
self.assertEquals(bc.get(BallColors.BLACK), 5)
self.assertEquals(bc.get(BallColors.WHITE), 1)
bc.removeOne(BallColors.BLACK)
self.assertEquals(bc.get(BallColors.BLACK), 4)
