add test_valid_key to utils

litellm/tests/test_utils.py

@@ -82,3 +82,24 @@ def test_get_valid_models():
     os.environ = old_environ
 
 # test_get_valid_models()
+
+
+
+
+# def test_bad_key():
+#     key = "bad-key"
+#     response = verify_access_key(key)
+#     if response == False:
+#         pass
+#     else:
+#         raise Exception("Bad key was not detected")
+# test_bad_key()
+
+# def test_good_key():
+#     key = os.environ['OPENAI_API_KEY']
+#     response = verify_access_key(key)
+#     if response == True:
+#         pass
+#     else:
+#         raise Exception("Good key did not pass")
+# test_good_key()

litellm/tests/test_verify_openai_key.py

@@ -1,20 +0,0 @@
-from litellm import verify_access_key
-import os
-
-def test_bad_key():
-    key = "bad-key"
-    response = verify_access_key(key)
-    if response == False:
-        pass
-    else:
-        raise Exception("Bad key was not detected")
-test_bad_key()
-
-def test_good_key():
-    key = os.environ['OPENAI_API_KEY']
-    response = verify_access_key(key)
-    if response == True:
-        pass
-    else:
-        raise Exception("Good key did not pass")
-test_good_key()