added tests

xflr6 · Mar 1, 2024 · 6890a51 · 6890a51
1 parent 1622675
commit 6890a51
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/concepts/examples.py b/concepts/examples.py
@@ -29,8 +29,8 @@ def load_dataset(name: str, data_src: typing.Optional[str] = DATASET_SOURCE,
 
     Example:
         >>> import concepts
-        >>> concepts.load_dataset('living_beings_and_water.cxt')
-        <Context object mapping 8 objects to 9 properties [b1e86589] at 0x...>
+        >>> concepts.load_dataset('livingbeings_en')
+        <Context object mapping 8 objects to 9 properties [e32693ba] at 0x...>
     """
 
     url = f"{data_src}/{name}.cxt"

diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -0,0 +1,10 @@
+import pytest
+
+import concepts
+
+
+def test_load_dataset():
+    context = concepts.load_dataset('livingbeings_en')
+
+    assert len(context.objects) == 8
+    assert len(context.properties) == 9