Add eq methods to IDView and IDStat

tests/conftest.py

@@ -249,6 +249,11 @@ def diedgelist2():
     return [({0, 1}, {2}), ({1, 2}, {4}), ({2, 3, 4}, {4, 5})]
 
 
+@pytest.fixture
+def diedgelist3():
+    return [({0}, {1, 2}), ({1}, {2, 4}), ({2, 3, 4}, {4, 5})]
+
+
 @pytest.fixture
 def diedgedict1():
     return {0: ({1, 2, 3}, {4}), 1: ({5, 6}, {6, 7, 8})}

tests/core/test_views.py

@@ -25,6 +25,25 @@ def test_neighbors(edgelist1, edgelist2, edgelist7):
     assert H3.edges.neighbors(3, s=2) == set()
 
 
+def test_hypergraph_eq(edgelist2, edgelist3, edgelist4, hyperwithattrs):
+    H2 = xgi.Hypergraph(edgelist2)
+    H3 = xgi.Hypergraph(edgelist3)
+    assert H2.nodes != H3.nodes
+    assert H2.edges != H3.edges
+
+    H4 = xgi.Hypergraph(edgelist4)
+    assert H4.nodes != hyperwithattrs.nodes
+    assert H4.edges == hyperwithattrs.edges
+
+
+def test_dihypergraph_eq(diedgelist2, diedgelist3, dihyperwithattrs):
+    H2 = xgi.DiHypergraph(diedgelist2)
+    H3 = xgi.DiHypergraph(diedgelist3)
+    assert H2.nodes != H3.nodes
+    assert H2.edges != H3.edges
+    assert H2.edges != dihyperwithattrs.edges
+
+
 def test_edge_order(edgelist3, diedgelist1):
     # undirected
     H = xgi.Hypergraph(edgelist3)

tests/stats/test_core_stats_functions.py

@@ -250,6 +250,18 @@ def test_dihypergraph_stats_items(diedgelist2):
         assert deg[n] == d
 
 
+def test_hypergraph_eq(edgelist2, edgelist3):
+    H2 = xgi.Hypergraph(edgelist2)
+    H3 = xgi.Hypergraph(edgelist3)
+    assert H2.nodes.degree != H3.nodes.degree
+
+
+def test_dihypergraph_eq(diedgelist2, diedgelist3):
+    H2 = xgi.DiHypergraph(diedgelist2)
+    H3 = xgi.DiHypergraph(diedgelist3)
+    assert H2.nodes.out_degree != H3.nodes.out_degree
+
+
 ### Numerical statistics
 
 

xgi/core/views.py

@@ -175,6 +175,27 @@ def __call__(self, bunch):
         """
         return self.from_view(self, bunch)
 
+    def __eq__(self, view):
+        """Checks whether the members and attributes of an IDView are equal.
+
+        Parameters
+        ----------
+        view : IDView
+            the IDView to be compared
+
+        Returns
+        -------
+        bool
+            Whether they are equal
+        """
+        return (
+            {idx: self._id_dict[idx] for idx in self._ids}
+            == {idx: view._id_dict[idx] for idx in view._ids}
+        ) and (
+            {idx: self._id_attr[idx] for idx in self._ids}
+            == {idx: view._id_attr[idx] for idx in view._ids}
+        )
+
     def filterby(self, stat, val, mode="eq"):
         """Filter the IDs in this view by a statistic.
 

xgi/stats/__init__.py

@@ -94,6 +94,9 @@ def __repr__(self):
     def __len__(self):
         return len(self.view)
 
+    def __eq__(self, stat):
+        return self.asdict() == stat.asdict()
+
     @property
     def name(self):
         """Name of this stat.
@@ -141,12 +144,12 @@ def asdict(self):
 
         """
         val = self._val
-        return {n: val[n] for n in self.view}
+        return {idx: val[idx] for idx in self.view}
 
     def aslist(self):
         """Output the stat as a list."""
         val = self._val
-        return [val[n] for n in self.view]
+        return [val[idx] for idx in self.view]
 
     def asnumpy(self):
         """Output the stat as a numpy array."""