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