Skip to content

Commit 8b67b07

Browse files
committed
Implement dt.Time type, and test built-in checks
Signed-off-by: Deepyaman Datta <[email protected]>
1 parent b915c5e commit 8b67b07

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

pandera/engines/ibis_engine.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,12 @@ def check(
6161
except TypeError:
6262
return False
6363

64+
def __str__(self) -> str:
65+
return str(self.type)
66+
67+
def __repr__(self) -> str:
68+
return f"DataType({self})"
69+
6470

6571
class Engine(
6672
metaclass=engine.Engine,
@@ -308,6 +314,21 @@ def from_parametrized_dtype(cls, ibis_dtype: dt.Timestamp):
308314
return cls(timezone=ibis_dtype.timezone, scale=ibis_dtype.scale)
309315

310316

317+
@Engine.register_dtype(
318+
equivalents=[
319+
"time",
320+
datetime.time,
321+
dt.Time,
322+
dt.time,
323+
]
324+
)
325+
@immutable
326+
class Time(DataType):
327+
"""Semantic representation of a :class:`dt.Time`."""
328+
329+
type = dt.time
330+
331+
311332
@Engine.register_dtype(
312333
equivalents=[
313334
"timedelta",

0 commit comments

Comments
 (0)