def test_init__roll_dies__1d2():
atk = Attack('bite', [2])
result = atk.roll_dies()
assert result >= 1
assert result <= 2
def test_init__roll_dies__2d1():
atk = Attack('bite', [1, 1])
result = atk.roll_dies()
assert result == 2
