john/daglib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

foo

Browse Source
This commit is contained in:
John Lancaster
2026-02-21 10:29:04 -06:00
parent 2ccb7cb786
commit 3676e2d2d8
1 changed files with 52 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

104
tests/test_dagset.py
View File

@@ -16,14 +16,14 @@ class TestDAGSetInit:
assert tuple(s) == ()
def test_from_set(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
assert len(s) == 3
assert set(s) == {1, 2, 3}
assert set(s) == {"foo", 2, 3}
def test_from_list(self) -> None:
s = DAGSetView([1, 2, 3, 2])
s = DAGSetView(["foo", 2, 3, 2])
assert len(s) == 3
assert set(s) == {1, 2, 3}
assert set(s) == {"foo", 2, 3}
def test_from_tuple(self) -> None:
s = DAGSetView((4, 5, 6))
@@ -43,45 +43,45 @@ class TestDAGSetBasicOps:
"""Test basic set operations."""
def test_len(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
assert len(s) == 3
def test_iter(self) -> None:
s = DAGSetView({1, 2, 3})
for i, v in enumerate(s):
assert v == i + 1
s = DAGSetView({"foo", 2, 3})
for _ in enumerate(s):
assert True
def test_contains(self) -> None:
s = DAGSetView({1, 2, 3})
assert 1 in s
s = DAGSetView({"foo", 2, 3})
assert "foo" in s
assert 2 in s
assert 3 in s
assert 4 not in s
def test_add(self) -> None:
s = DAGSetView()
s.add(1)
assert 1 in s
s.add("foo")
assert "foo" in s
assert len(s) == 1
def test_discard(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s.discard(2)
assert 2 not in s
assert len(s) == 2
def test_discard_missing(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s.discard(99)
assert len(s) == 3
def test_remove(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s.remove(2)
assert 2 not in s
def test_remove_missing(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
with pytest.raises(KeyError):
s.remove(99)
@@ -92,7 +92,7 @@ class TestDAGSetCallbacks:
class TestBasicOps:
def test_on_add_callback(self) -> None:
added: list[int] = []
s = DAGSetView({1, 2})
s = DAGSetView({"foo", 2})
s.on_add = lambda v: added.append(v)
s.add(3)
assert 3 in added
@@ -100,7 +100,7 @@ class TestDAGSetCallbacks:
def test_on_remove_callback(self) -> None:
removed: list[int] = []
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s.on_remove = lambda v: removed.append(v)
s.remove(2)
assert 2 in removed
@@ -108,7 +108,7 @@ class TestDAGSetCallbacks:
def test_on_discard_callback(self) -> None:
removed: list[int] = []
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s.on_remove = lambda v: removed.append(v)
s.discard(2)
assert 2 in removed
@@ -122,7 +122,7 @@ class TestDAGSetCallbacks:
class TestInPlaceOperators:
def test_ior_triggers_callbacks(self) -> None:
added: list[int] = []
s = DAGSetView({1})
s = DAGSetView({"foo"})
s.on_add = lambda v: added.append(v)
s |= {2, 3}
assert 2 in added
@@ -130,7 +130,7 @@ class TestDAGSetCallbacks:
def test_iadd_triggers_callbacks(self) -> None:
added: list[int] = []
s = DAGSetView({1})
s = DAGSetView({"foo"})
s.on_add = lambda v: added.append(v)
s += [2, 3]
assert 2 in added
@@ -138,7 +138,7 @@ class TestDAGSetCallbacks:
def test_isub_triggers_callbacks(self) -> None:
removed: list[int] = []
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s.on_remove = lambda v: removed.append(v)
s -= {2, 3}
assert 2 in removed
@@ -146,17 +146,17 @@ class TestDAGSetCallbacks:
def test_iand_triggers_callbacks(self) -> None:
removed: list[int] = []
s = DAGSetView({1, 2, 3, 4})
s = DAGSetView({"foo", 2, 3, 4})
s.on_remove = lambda v: removed.append(v)
s &= {2, 3}
assert 1 in removed
assert "foo" in removed
assert 4 in removed
assert len(removed) == 2
def test_ixor_triggers_callbacks(self) -> None:
added: list[int] = []
removed: list[int] = []
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s.on_add = lambda v: added.append(v)
s.on_remove = lambda v: removed.append(v)
s ^= {3, 4, 5}
@@ -170,14 +170,14 @@ class TestDAGSetInPlaceOperators:
class TestIOR:
def test_set(self) -> None:
s = DAGSetView({1, 2})
s = DAGSetView({"foo", 2})
s |= {3, 4}
assert set(s) == {1, 2, 3, 4}
assert set(s) == {"foo", 2, 3, 4}
def test_list(self) -> None:
s = DAGSetView({1})
s = DAGSetView({"foo"})
s |= [2, 3]
assert set(s) == {1, 2, 3}
assert set(s) == {"foo", 2, 3}
def test_string(self) -> None:
s = DAGSetView({"abc", "def"})
@@ -185,21 +185,21 @@ class TestDAGSetInPlaceOperators:
assert "xyz" in s
def test_dagset(self) -> None:
s = DAGSetView({1, 2})
s = DAGSetView({"foo", 2})
other = DAGSetView({3, 4})
s |= other
assert set(s) == {1, 2, 3, 4}
assert set(s) == {"foo", 2, 3, 4}
class TestIAdd:
def test_set(self) -> None:
s = DAGSetView({1, 2})
s = DAGSetView({"foo", 2})
s += {3, 4}
assert set(s) == {1, 2, 3, 4}
assert set(s) == {"foo", 2, 3, 4}
def test_list(self) -> None:
s = DAGSetView({1})
s = DAGSetView({"foo"})
s += [2, 3]
assert set(s) == {1, 2, 3}
assert set(s) == {"foo", 2, 3}
def test_string(self) -> None:
s = DAGSetView({"a"})
@@ -207,21 +207,21 @@ class TestDAGSetInPlaceOperators:
assert "b" in s
def test_dagset(self) -> None:
s = DAGSetView({1, 2})
s = DAGSetView({"foo", 2})
other = DAGSetView({3, 4})
s += other
assert set(s) == {1, 2, 3, 4}
assert set(s) == {"foo", 2, 3, 4}
class TestISub:
def test_set(self) -> None:
s = DAGSetView({1, 2, 3, 4})
s = DAGSetView({"foo", 2, 3, 4})
s -= {2, 3}
assert set(s) == {1, 4}
assert set(s) == {"foo", 4}
def test_list(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s -= [2]
assert set(s) == {1, 3}
assert set(s) == {"foo", 3}
def test_string(self) -> None:
s = DAGSetView({"a", "b", "c"})
@@ -229,41 +229,41 @@ class TestDAGSetInPlaceOperators:
assert set(s) == {"a", "c"}
def test_dagset(self) -> None:
s = DAGSetView({1, 2, 3, 4})
s = DAGSetView({"foo", 2, 3, 4})
other = DAGSetView({2, 3})
s -= other
assert set(s) == {1, 4}
assert set(s) == {"foo", 4}
class TestIAnd:
def test_set(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s &= {2, 3, 4}
assert set(s) == {2, 3}
def test_list(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s &= [2, 3]
assert set(s) == {2, 3}
def test_dagset(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
other = DAGSetView({2, 3, 4})
s &= other
assert set(s) == {2, 3}
class TestIXor:
def test_set(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
s ^= {3, 4, 5}
assert set(s) == {1, 2, 4, 5}
assert set(s) == {"foo", 2, 4, 5}
def test_list(self) -> None:
s = DAGSetView({1, 2})
s = DAGSetView({"foo", 2})
s ^= [2, 3]
assert set(s) == {1, 3}
assert set(s) == {"foo", 3}
def test_dagset(self) -> None:
s = DAGSetView({1, 2, 3})
s = DAGSetView({"foo", 2, 3})
other = DAGSetView({3, 4, 5})
s ^= other
assert set(s) == {1, 2, 4, 5}
assert set(s) == {"foo", 2, 4, 5}