File size: 686 Bytes
767ca1c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from os import getenv
from unittest import TestCase
import openai
from osbot_utils.utils.Dev import pprint
from test_bot.api.Open_API import Open_API, OPEN_API_KEY
class test_Open_API(TestCase):
def setUp(self) -> None:
self.open_api = Open_API().setup()
def test___init__(self):
assert type(self.open_api) == Open_API
def test_api_key(self):
api_key = self.open_api.api_key()
assert api_key is not None
assert api_key == getenv(OPEN_API_KEY)
def test_create(self):
response = self.open_api.create()
pprint(list(response))
def test_setup(self):
assert openai.api_key == self.open_api.api_key() |