File size: 857 Bytes
bd91486 | 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 29 30 31 32 | import pytest
from cogenbai.core.model import CogenBAI
def test_model_initialization():
model = CogenBAI()
assert model is not None
assert model.device in ["cuda", "cpu"]
def test_code_generation():
model = CogenBAI()
prompt = "def calculate_fibonacci(n):"
result = model.generate_code(prompt)
assert isinstance(result, str)
assert len(result) > 0
import unittest
from cogenbai.core.model import CogenBAI
class TestCogenBAI(unittest.TestCase):
def setUp(self):
self.model = CogenBAI()
def test_generate_code(self):
prompt = "Create a function that adds two numbers"
language = "python"
result = self.model.generate_code(prompt, language)
self.assertIsInstance(result, str)
self.assertGreater(len(result), 0)
if __name__ == '__main__':
unittest.main()
|