medium_test_code_generation

test_unit_sample.py

def test_pydantic_models_generator_basic():

    ddl = """
CREATE table user_history (
        runid                 decimal(21) null
    ) ;
"""
    result = create_models(ddl, models_type="pydantic")["code"]

    expected = """from typing import Optional
from pydantic import BaseModel


class UserHistory(BaseModel):

    runid: Optional[float]
"""
    assert result == expected