镜像自地址
https://github.com/binary-husky/gpt_academic.git
已同步 2025-12-06 14:36:48 +00:00
63 行
3.0 KiB
Python
63 行
3.0 KiB
Python
import unittest
|
|
|
|
def validate_path():
|
|
import os, sys
|
|
|
|
os.path.dirname(__file__)
|
|
root_dir_assume = os.path.abspath(os.path.dirname(__file__) + "/..")
|
|
os.chdir(root_dir_assume)
|
|
sys.path.append(root_dir_assume)
|
|
|
|
|
|
validate_path() # validate path so you can run from base directory
|
|
|
|
from shared_utils.key_pattern_manager import is_openai_api_key
|
|
|
|
class TestKeyPatternManager(unittest.TestCase):
|
|
def test_is_openai_api_key_with_valid_key(self):
|
|
key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
|
|
self.assertTrue(is_openai_api_key(key))
|
|
|
|
key = "sx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
|
|
self.assertFalse(is_openai_api_key(key))
|
|
|
|
key = "sess-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
|
|
self.assertTrue(is_openai_api_key(key))
|
|
|
|
key = "sess-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
|
|
self.assertFalse(is_openai_api_key(key))
|
|
|
|
key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx_xxxxxxxxxxxxxxxxxx-xxxxxxxxxxxxxxxxxxxx"
|
|
self.assertTrue(is_openai_api_key(key))
|
|
key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx-xxxxxxxxxxxxxxx_xxxxxxxxxxxx_xxxxx-xxxxxxxxxxxxxxxxxxxx"
|
|
self.assertTrue(is_openai_api_key(key))
|
|
key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx-xxx-xxxxxxxxxxxxxxx_xxxxxxxxxxxx_xxxxx-xxxxxx-xxxxxxxxxxxxx"
|
|
self.assertTrue(is_openai_api_key(key))
|
|
key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx-xxx-xxxxxxxxxxxxxxx_xxxxxxxxxxxx_xxxxx-xxxxxxxxxxxxxxxxxx"
|
|
self.assertFalse(is_openai_api_key(key))
|
|
key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx-xxx-xxxxxxxxxxxxxxx_xxxxxxxxxxxx_xxxxx-xxxxxxxxxxxxxxxxxxxxx"
|
|
self.assertFalse(is_openai_api_key(key))
|
|
|
|
key = "sk-proj-xx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx_xxxxxxxxxxxxxxxxxxxxx-xxxxxxxx"
|
|
self.assertTrue(is_openai_api_key(key))
|
|
key = "sk-proj-xx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx_xxxxxxxxxxxx_xxxxxxxxxxxxxxxxxxxxx-xxxxxxxx"
|
|
self.assertTrue(is_openai_api_key(key))
|
|
key = "sk-proj-xx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx_xxxxxxxxxxxx_xxxxxxxxxxxxxxxxxx-xxxxxxxx"
|
|
self.assertFalse(is_openai_api_key(key))
|
|
key = "sk-proj-xx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx_xxxxxxxxxxxx_xxxxxxxxxxxxxxxxxx-xxxxxxxxxxxxx"
|
|
self.assertFalse(is_openai_api_key(key))
|
|
key = "sk-proj-xx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx_xxxxxxxxxxxx_xxxxxxxxxxxxxxxxxx-xxx啊xxxxxxx"
|
|
self.assertFalse(is_openai_api_key(key))
|
|
|
|
|
|
def test_is_openai_api_key_with_invalid_key(self):
|
|
key = "invalid_key"
|
|
self.assertFalse(is_openai_api_key(key))
|
|
|
|
def test_is_openai_api_key_with_custom_pattern(self):
|
|
# Assuming you have set a custom pattern in your configuration
|
|
key = "custom-pattern-key"
|
|
self.assertFalse(is_openai_api_key(key))
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |