fixing tests for overflows

CAVEconnectome · Dec 9, 2024 · 5a6988b · 5a6988b
1 parent a0846fc
commit 5a6988b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/caveclient/chunkedgraph.py b/caveclient/chunkedgraph.py
@@ -1317,7 +1317,13 @@ def is_valid_nodes(
         -------
         np.array of bool
             Array of whether these are valid IDs.
+
+        Raises
+        -----
+        OverflowError
+            If the root_id are too large or negative to be represented as np.uint64.
         """
+
         node_ids = root_id_int_list_check(node_ids, make_unique=False)
 
         endpoint_mapping = self.default_url_mapping

diff --git a/tests/test_chunkedgraph.py b/tests/test_chunkedgraph.py
@@ -1037,5 +1037,5 @@ def test_is_valid_nodes(self, myclient):
             json=return_data,
             match=[json_params_matcher(data)],
         )
-        out = myclient.chunkedgraph.is_valid_nodes(query_nodes)
-        assert not np.any(out)
+        with pytest.raises(OverflowError):
+            out = myclient.chunkedgraph.is_valid_nodes(query_nodes)