manny405 / sapai

Super auto pets engine built with reinforment learning training in mind
MIT License
66 stars 21 forks source link

Cupcake & Horse stats buff is permanent #50

Closed alexdriedger closed 2 years ago

alexdriedger commented 2 years ago

These tests currently fail. It looks like untilEndOfBattle is in the data for multiple pets (and cupcake) but the code doesn't do anything with it

def test_cupcake(self):
    player = Player(shop=Shop(["cupcake"]), team=Team([Pet("fish")]))

    player.buy_food(0, 0)
    self.assertEqual(player.team[0].attack, 5)
    self.assertEqual(player.team[0].health, 6)

    player.end_turn()
    player.start_turn()

    self.assertEqual(player.team[0].attack, 2)
    self.assertEqual(player.team[0].health, 3)
def test_horse(self):
    player = Player(shop=Shop(["fish"]), team=Team([Pet("horse")]))

    player.buy_pet(0)
    self.assertEqual(player.team[1].attack, 3)
    self.assertEqual(player.team[1].health, 3)

    player.end_turn()
    player.start_turn()

    self.assertEqual(player.team[1].attack, 2)
    self.assertEqual(player.team[1].health, 3)