test: add example test

genai-impact · Feb 27, 2024 · d7c2bd7 · d7c2bd7
1 parent 063f858
commit d7c2bd7
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/genai_impact/__init__.py b/genai_impact/__init__.py
@@ -1 +1 @@
-from client_wrapper import OpenAI
+from .client_wrapper import OpenAI
diff --git a/tests/test_compute_impacts.py b/tests/test_compute_impacts.py
@@ -0,0 +1,12 @@
+import pytest
+
+from genai_impact.compute_impacts import compute_llm_impact
+
+
+@pytest.mark.parametrize('model_size,output_tokens', [
+    (130, 1000),
+    (7, 150)
+])
+def test_compute_impacts_is_positive(model_size: float, output_tokens: int) -> None:
+    impacts = compute_llm_impact(model_size, output_tokens)
+    assert impacts.energy >= 0
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		from client_wrapper import OpenAI
		from .client_wrapper import OpenAI