dogeystamp/bitmask
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

bitmask.py: fix unit tests

Browse Source
This commit is contained in:
dogeystamp 2022-08-03 14:58:25 -04:00
parent 060d91473b
commit 27e4b6f6fe
Signed by: dogeystamp
GPG Key ID: 7225FE3592EFFA38
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_bitmask.py
View File

@ -24,8 +24,8 @@ def test_eq():
# Equality # Equality
assert Bitmask() == Bitmask() assert Bitmask() == Bitmask()
assert Bitmask(Desc.SMALL) == Bitmask(Desc.SMALL) assert Bitmask(Desc.SMALL) == Bitmask(Desc.SMALL)
assert Bitmask(Desc.SMALL, Desc.ROUND) == Bitmask(Desc, Desc.SMALL, Desc.ROUND) assert Bitmask(Desc.SMALL, Desc.ROUND) == Bitmask(Desc.SMALL, Desc.ROUND)
assert Bitmask(Desc.ROUND, Desc.SMALL) == Bitmask(Desc, Desc.SMALL, Desc.ROUND) assert Bitmask(Desc.ROUND, Desc.SMALL) == Bitmask(Desc.SMALL, Desc.ROUND)
# Inequality # Inequality
assert Bitmask(Desc.SMALL) != Bitmask(Desc.SMALL, Desc.ROUND) assert Bitmask(Desc.SMALL) != Bitmask(Desc.SMALL, Desc.ROUND)
@ -38,7 +38,7 @@ def test_eq():
# Wrong types # Wrong types
assert Bitmask() != "Hello World!" assert Bitmask() != "Hello World!"
assert Bitmask() != 0 assert Bitmask() != 0
assert Bitmask() != Bitmask(Colors) assert Bitmask(Desc.SMALL) != Bitmask(Colors.TEAL)
def test_repr(): def test_repr():
@ -62,7 +62,7 @@ def test_add():
mask.add(Desc.ROUND) mask.add(Desc.ROUND)
assert mask == Bitmask(Desc.ROUND) assert mask == Bitmask(Desc.ROUND)
with pytest.raises(TypeError, match=".* Desc .* Bitmask.*"): with pytest.raises(TypeError, match=".* Bitmask .* Desc .* Bitmask.*"):
mask.add(1) mask.add(1)
with pytest.raises(TypeError, match=r".* Bitmask .* Desc .* Bitmask.*"): with pytest.raises(TypeError, match=r".* Bitmask .* Desc .* Bitmask.*"):
mask += 1 mask += 1
@ -192,22 +192,22 @@ def test_discard():
empty_mask = Bitmask() empty_mask = Bitmask()
empty_mask.discard(Desc.SMALL) empty_mask.discard(Desc.SMALL)
assert empty_mask == Bitmask() assert empty_mask == Bitmask()
with pytest.raises(TypeError, match="can only discard Desc from Bitmask"): with pytest.raises(TypeError, match=".* Bitmask or Desc from Bitmask"):
empty_mask.discard(Bitmask(Desc.SMALL)) empty_mask.discard(Bitmask(Desc.SMALL))
def test_subtract(): def test_subtract():
"""Test - operator.""" """Test - operator."""
# Individual flag # Individual flag
assert Bitmask(Desc.SMALL, Desc.FUNKY) - Desc.SMALL == Bitmask(Desc, Desc.FUNKY) assert Bitmask(Desc.SMALL, Desc.FUNKY) - Desc.SMALL == Bitmask(Desc.FUNKY)
assert Bitmask(Desc.FUNKY, Desc.SMALL) - Desc.SMALL == Bitmask(Desc, Desc.FUNKY) assert Bitmask(Desc.FUNKY, Desc.SMALL) - Desc.SMALL == Bitmask(Desc.FUNKY)
# Two bitmasks # Two bitmasks
assert Bitmask(Desc.SMALL, Desc.FUNKY, Desc.ROUND) - Bitmask( assert Bitmask(Desc.SMALL, Desc.FUNKY, Desc.ROUND) - Bitmask(
Desc, Desc.SMALL, Desc.ROUND Desc, Desc.SMALL, Desc.ROUND
) == Bitmask(Desc.FUNKY) ) == Bitmask(Desc.FUNKY)
assert ( assert (
Bitmask(Desc.FUNKY, Desc.SMALL) - Bitmask(Desc, Desc.SMALL, Desc.FUNKY) Bitmask(Desc.FUNKY, Desc.SMALL) - Bitmask(Desc.SMALL, Desc.FUNKY)
== Bitmask() == Bitmask()
) )
@ -233,7 +233,7 @@ def test_xor_operator():
Desc, Desc.SMALL, Desc.FUNKY, Desc.ROUND Desc, Desc.SMALL, Desc.FUNKY, Desc.ROUND
) )
assert ( assert (
Bitmask(Desc.FUNKY, Desc.ROUND) ^ Bitmask(Desc, Desc.FUNKY, Desc.ROUND) Bitmask(Desc.FUNKY, Desc.ROUND) ^ Bitmask(Desc.FUNKY, Desc.ROUND)
== Bitmask() == Bitmask()
) )
assert Bitmask() ^ Bitmask() == Bitmask() assert Bitmask() ^ Bitmask() == Bitmask()