class PatterncraftVisitorTestCase(unittest.TestCase):
    def setUp(self):
        self.bullet = TankBullet()
        self.light = Marine()
        self.bullet2 = TankBullet()
        self.armored = Marauder()

    def test_equals(self):
        self.light.accept(self.bullet)
        self.assertEqual(self.light.health, 100 - 21)

    def test_equals_2(self):
        self.armored.accept(self.bullet2)
        self.assertEqual(self.armored.health, 125 - 32)
 def setUp(self):
     self.bullet = TankBullet()
     self.light = Marine()
     self.bullet2 = TankBullet()
     self.armored = Marauder()