rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-01-07 20:32:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add pyarrow date and timestamp type conversion from pydantic (#576)

Browse Source
This commit is contained in:
Prashanth Rao
2023-10-16 22:42:24 -04:00
committed by GitHub
parent bb01ad5290
commit 86efb11572
2 changed files with 28 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
python/tests/test_pydantic.py
View File

@@ -14,6 +14,7 @@
import json
import sys
from datetime import date, datetime
from typing import List, Optional
import pyarrow as pa
@@ -40,10 +41,18 @@ def test_pydantic_to_arrow():
li: List[int]
opt: Optional[str] = None
st: StructModel
dt: date
dtt: datetime
# d: dict
m = TestModel(
id=1, s="hello", vec=[1.0, 2.0, 3.0], li=[2, 3, 4], st=StructModel(a="a", b=1.0)
id=1,
s="hello",
vec=[1.0, 2.0, 3.0],
li=[2, 3, 4],
st=StructModel(a="a", b=1.0),
dt=date.today(),
dtt=datetime.now(),
)
schema = pydantic_to_schema(TestModel)
@@ -62,6 +71,8 @@ def test_pydantic_to_arrow():
),
False,
),
pa.field("dt", pa.date32(), False),
pa.field("dtt", pa.timestamp("us"), False),
]
)
assert schema == expect_schema
@@ -79,10 +90,18 @@ def test_pydantic_to_arrow_py38():
li: List[int]
opt: Optional[str] = None
st: StructModel
dt: date
dtt: datetime
# d: dict
m = TestModel(
id=1, s="hello", vec=[1.0, 2.0, 3.0], li=[2, 3, 4], st=StructModel(a="a", b=1.0)
id=1,
s="hello",
vec=[1.0, 2.0, 3.0],
li=[2, 3, 4],
st=StructModel(a="a", b=1.0),
dt=date.today(),
dtt=datetime.now(),
)
schema = pydantic_to_schema(TestModel)
@@ -101,6 +120,8 @@ def test_pydantic_to_arrow_py38():
),
False,
),
pa.field("dt", pa.date32(), False),
pa.field("dtt", pa.timestamp("us"), False),
]
)
assert schema == expect_schema