update tests

SubstrateLabs · Apr 30, 2024 · 1eabe26 · 1eabe26
1 parent 4e98bc1
commit 1eabe26
Show file tree

Hide file tree

Showing 10 changed files with 35 additions and 75 deletions.
diff --git a/tests/pydantic-1/poetry.lock b/tests/pydantic-1/poetry.lock
diff --git a/tests/pydantic-1/pyproject.toml b/tests/pydantic-1/pyproject.toml
@@ -8,7 +8,9 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.10"
 pydantic = "<2.0.0"
-substrate = "*"
+networkx = "^3.3"
+httpx = "^0.27.0"
+distro = "^1.9.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.2.0"

diff --git a/tests/pydantic-1/tests/test_basic.py b/tests/pydantic-1/tests/test_basic.py
@@ -1,8 +1,14 @@
 import os
+import sys
+from pathlib import Path
 
 
 class Test:
     def test_basic(self):
+        # add parent dir to sys.path to make 'substrate' importable
+        parent_dir = Path(__file__).resolve().parent.parent.parent.parent
+        sys.path.insert(0, str(parent_dir))
+
         api_key = os.environ.get("SUBSTRATE_API_KEY")
         if api_key is None:
             raise EnvironmentError("No SUBSTRATE_API_KEY set")
@@ -13,6 +19,9 @@ def test_basic(self):
 
         story = GenerateText({"prompt": "tell me a story", "max_tokens": 8})
         response = substrate.run(story)
+        import json
+
+        print("response", json.dumps(response.json, indent=2))
 
         summary_out = response.get(story)
         assert len(summary_out.text) > 0
diff --git a/tests/pydantic-2/basic.py b/tests/pydantic-2/basic.py
diff --git a/tests/pydantic-2/poetry.lock b/tests/pydantic-2/poetry.lock
diff --git a/tests/pydantic-2/pyproject.toml b/tests/pydantic-2/pyproject.toml
@@ -8,7 +8,9 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.10"
 pydantic = "^2.0.0"
-substrate = "*"
+networkx = "^3.3"
+httpx = "^0.27.0"
+distro = "^1.9.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.2.0"

diff --git a/tests/pydantic-2/tests/test_basic.py b/tests/pydantic-2/tests/test_basic.py
@@ -1,8 +1,14 @@
 import os
+import sys
+from pathlib import Path
 
 
 class Test:
     def test_basic(self):
+        # add parent dir to sys.path to make 'substrate' importable
+        parent_dir = Path(__file__).resolve().parent.parent.parent.parent
+        sys.path.insert(0, str(parent_dir))
+
         api_key = os.environ.get("SUBSTRATE_API_KEY")
         if api_key is None:
             raise EnvironmentError("No SUBSTRATE_API_KEY set")

diff --git a/tests/python-3-9/poetry.lock b/tests/python-3-9/poetry.lock
diff --git a/tests/python-3-9/pyproject.toml b/tests/python-3-9/pyproject.toml
@@ -7,7 +7,11 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "^3.9"
-substrate = "*"
+typing-extensions = "^4.11.0"
+pydantic = "^2.7.1"
+networkx = "3.2.1"
+httpx = "^0.27.0"
+distro = "^1.9.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.2.0"

diff --git a/tests/python-3-9/tests/test_basic.py b/tests/python-3-9/tests/test_basic.py
@@ -1,8 +1,14 @@
 import os
+import sys
+from pathlib import Path
 
 
 class Test:
     def test_basic(self):
+        # add parent dir to sys.path to make 'substrate' importable
+        parent_dir = Path(__file__).resolve().parent.parent.parent.parent
+        sys.path.insert(0, str(parent_dir))
+
         api_key = os.environ.get("SUBSTRATE_API_KEY")
         if api_key is None:
             raise EnvironmentError("No SUBSTRATE_API_KEY set")