From ff36b9c6dadad632c36de4bff0b86287ac35c65c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 12 Sep 2023 11:45:55 -0700 Subject: [PATCH] add test_valid_key to utils --- litellm/tests/test_utils.py | 23 ++++++++++++++++++++++- litellm/tests/test_verify_openai_key.py | 20 -------------------- 2 files changed, 22 insertions(+), 21 deletions(-) delete mode 100644 litellm/tests/test_verify_openai_key.py diff --git a/litellm/tests/test_utils.py b/litellm/tests/test_utils.py index dfe40d5fa..ed482b965 100644 --- a/litellm/tests/test_utils.py +++ b/litellm/tests/test_utils.py @@ -81,4 +81,25 @@ def test_get_valid_models(): # reset replicate env key os.environ = old_environ -# test_get_valid_models() \ No newline at end of file +# 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() \ No newline at end of file diff --git a/litellm/tests/test_verify_openai_key.py b/litellm/tests/test_verify_openai_key.py deleted file mode 100644 index e671641ac..000000000 --- a/litellm/tests/test_verify_openai_key.py +++ /dev/null @@ -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() \ No newline at end of file