#!/usr/bin/python

import unittest

import main



class ExamplesPart1(unittest.TestCase):
    def test_example1(self):
        player = main.Entity()
        player.health =  10
        player.armour =   0
        player.mana   = 250

        boss = main.Entity()
        boss.health = 13
        boss.damage =  8

        result = main.solve_partn(1, player, boss)
        self.assertEqual(result, 173 + 53)

    def test_example2(self):
        player = main.Entity()
        player.health =  10
        player.armour =   0
        player.mana   = 250

        boss = main.Entity()
        boss.health = 14
        boss.damage =  8

        result = main.solve_partn(1, player, boss)
        self.assertEqual(result, 229 + 113 + 73 + 173 + 53)



if __name__ == "__main__":
    unittest.main()