Skip to content

Commit

Permalink
Add metronome flag in move flag unit test
Browse files Browse the repository at this point in the history
  • Loading branch information
hsahovic committed Jan 15, 2024
1 parent a6308a6 commit 66d9502
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions unit_tests/environment/test_move.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,8 +161,9 @@ def test_flags():
flame_thrower = Move("flamethrower", gen=8)
sludge_bomb = Move("sludgebomb", gen=8)

assert flame_thrower.flags == {"protect", "mirror"}
assert sludge_bomb.flags == {"bullet", "protect", "mirror"}
print(flame_thrower.flags)
assert flame_thrower.flags == {"metronome", "protect", "mirror"}
assert sludge_bomb.flags == {"bullet", "metronome", "protect", "mirror"}
for move in move_generator():
assert isinstance(move.flags, set)

Expand Down

0 comments on commit 66d9502

Please sign in to comment.