Merge pull request #63 from edgenai/chore/issue9

Chore/issue9

tests/test_audio.py

@@ -27,13 +27,26 @@ def test_transcriptions():
     print(transcription)
     print(expected)
 
-    assert(type(transcription) is str)
+    have = transcription.text
 
-    d = Levenshtein.distance(transcription, expected)
+    assert(type(have) is str)
+
+    d = Levenshtein.distance(have, expected)
     similarity = 100 - ((d / len(expected)) * 100)
-    print(f"distance: {d} of '{transcription}', similarity: {similarity}")
+    print(f"distance: {d} of '{have}', similarity: {similarity}")
 
     assert(similarity > 90.0)
 
+def test_transcriptions_status():
+    try:
+        status = client.audio.transcriptions.status.create()
+    except APIConnectionError:
+        pytest.fail("No connection. Is edgen running?")
+
+    model = status.active_model
+    assert(type(model) is str)
+    print(model)
+
 if __name__ == "__main__":
    test_transcriptions()
+   test_transcriptions_status()

tests/test_chat.py

@@ -50,5 +50,16 @@ def test_completions():
     assert(type(content) is str)
     assert("3" in content)
 
+def test_completions_status():
+    try:
+        status = client.chat.completions.status.create()
+    except APIConnectionError:
+        pytest.fail("No connection. Is edgen running?")
+
+    model = status.active_model
+    assert(type(model) is str)
+    print(model)
+
 if __name__ == "__main__":
    test_completions_streaming()
+   test_completions()

tests/test_misc.py

@@ -20,7 +20,7 @@ def test_version():
     assert(format_version(version) == expected)
 
 def edgen_version():
-    finished = subprocess.run(["cargo", "run", "version"], capture_output=True, text=True)
+    finished = subprocess.run(["target/release/edgen", "version"], capture_output=True, text=True)
     version = ''.join(f"{finished.stdout}".splitlines())
     return version