Update tests for the relations/annotations API

Replace assertions on the removed interactions/associations endpoints
and the dropped direction/sign enum filters with checks against the
new exports/relations/parquet endpoint and its real parameters.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: deeenes

tests/test_inventory.py

@@ -27,8 +27,8 @@ def test_builds_endpoints(self):
 
         assert len(endpoints) > 0
         assert 'exports/entities/parquet' in endpoints
-        assert 'exports/interactions/parquet' in endpoints
-        assert 'exports/associations/parquet' in endpoints
+        assert 'exports/relations/parquet' in endpoints
+        assert 'exports/annotations/parquet' in endpoints
 
     def test_entity_filters(self):
 
@@ -41,18 +41,15 @@ def test_entity_filters(self):
         assert 'taxonomy_ids' in ep.params
         assert 'entity_types' in ep.params
 
-    def test_interaction_filters(self):
+    def test_relation_filters(self):
 
         endpoints = _build_static_fallback()
-        ep = endpoints['exports/interactions/parquet']
+        ep = endpoints['exports/relations/parquet']
 
-        assert 'direction' in ep.params
-        pdef = ep.params['direction']
-        assert pdef.allowed_values == [
-            'any',
-            'directed',
-            'undirected',
-        ]
+        assert 'subject_entity_pks' in ep.params
+        assert 'object_entity_pks' in ep.params
+        assert 'predicates' in ep.params
+        assert 'relation_categories' in ep.params
 
 
 class TestParseOpenapi:
@@ -72,8 +69,8 @@ def test_parse_local_spec(self):
 
         assert len(endpoints) > 0
         assert 'exports/entities/parquet' in endpoints
-        assert 'exports/interactions/parquet' in endpoints
-        assert 'exports/associations/parquet' in endpoints
+        assert 'exports/relations/parquet' in endpoints
+        assert 'exports/annotations/parquet' in endpoints
 
     def test_export_endpoints_have_filters(self):
 
@@ -84,19 +81,12 @@ def test_export_endpoints_have_filters(self):
             spec = json.load(f)
 
         endpoints = parse_openapi(spec)
-        ep = endpoints['exports/interactions/parquet']
+        ep = endpoints['exports/relations/parquet']
 
         assert ep.method == 'POST'
         assert ep.response_format == 'parquet'
-        assert 'entity_ids' in ep.params
-        assert 'direction' in ep.params
-
-        direction = ep.params['direction']
-        assert direction.allowed_values == [
-            'any',
-            'directed',
-            'undirected',
-        ]
+        assert 'sources' in ep.params
+        assert 'predicates' in ep.params
 
     def test_ontology_endpoints(self):
 
@@ -131,22 +121,23 @@ def test_params_method(self):
         inv = Inventory(base_url='http://localhost:99999')
         inv.load()
 
-        params = inv.params('exports/interactions/parquet')
+        params = inv.params('exports/relations/parquet')
 
-        assert 'entity_ids' in params
-        assert isinstance(params['entity_ids'], ParamDef)
+        assert 'sources' in params
+        assert isinstance(params['sources'], ParamDef)
 
-    def test_allowed_values_method(self):
+    def test_allowed_values_unconstrained_relation_filter(self):
 
         inv = Inventory(base_url='http://localhost:99999')
         inv.load()
 
         values = inv.allowed_values(
-            'exports/interactions/parquet',
-            'direction',
+            'exports/relations/parquet',
+            'sources',
         )
 
-        assert values == ['any', 'directed', 'undirected']
+        # The static fallback does not enumerate allowed sources.
+        assert values is None
 
     def test_allowed_values_unconstrained(self):
 

tests/test_query.py

@@ -8,7 +8,6 @@
 from omnipath_client._errors import (
     UnknownEndpointError,
     UnknownParameterError,
-    InvalidParameterValueError,
 )
 from omnipath_client._inventory import Inventory
 
@@ -39,16 +38,19 @@ def test_valid_entity_query(self, query_builder):
         assert query.endpoint.path == '/exports/entities/parquet'
         assert query.params == {'taxonomy_ids': ['9606']}
 
-    def test_valid_interaction_query(self, query_builder):
+    def test_valid_relation_query(self, query_builder):
 
         query = query_builder.build(
-            'exports/interactions/parquet',
-            entity_ids=['Q9Y6K9'],
-            direction='directed',
+            'exports/relations/parquet',
+            entity_pks=['12345'],
+            sources=['signor'],
+            predicates=['interacts_with'],
         )
 
-        assert query.params['entity_ids'] == ['Q9Y6K9']
-        assert query.params['direction'] == 'directed'
+        assert query.endpoint.path == '/exports/relations/parquet'
+        assert query.params['entity_pks'] == ['12345']
+        assert query.params['sources'] == ['signor']
+        assert query.params['predicates'] == ['interacts_with']
 
     def test_unknown_endpoint(self, query_builder):
 
@@ -63,23 +65,6 @@ def test_unknown_parameter(self, query_builder):
                 bogus_param='value',
             )
 
-    def test_invalid_enum_value(self, query_builder):
-
-        with pytest.raises(InvalidParameterValueError):
-            query_builder.build(
-                'exports/interactions/parquet',
-                direction='sideways',
-            )
-
-    def test_valid_enum_value(self, query_builder):
-
-        query = query_builder.build(
-            'exports/interactions/parquet',
-            sign='positive',
-        )
-
-        assert query.params['sign'] == 'positive'
-
     def test_none_values_skipped(self, query_builder):
 
         query = query_builder.build(