add check

This commit is contained in:
Jason Liu
2023-10-14 16:04:10 -04:00
parent 2bb47034c2
commit bc2c6f9fa6
2 changed files with 20 additions and 1 deletions
+17 -1
View File
@@ -1,4 +1,4 @@
from pyexpat import model
import pytest
import openai
from pydantic import BaseModel
from instructor.distil import (
@@ -19,6 +19,22 @@ class SimpleModel(BaseModel):
data: int
def test_must_have_hint():
with pytest.raises(AssertionError):
@instructions.distil
def test_func(x: int):
return SimpleModel(data=x)
def test_must_be_base_model():
with pytest.raises(AssertionError):
@instructions.distil
def test_func(x) -> int:
return SimpleModel(data=x)
def test_is_return_type_base_model_or_instance():
def valid_function() -> SimpleModel:
return SimpleModel(data=1)