Skip to content

Commit d5f671c

Browse files
committed
Fix faulty testconf
1 parent 350cbc1 commit d5f671c

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

bindings/python/tests/conftest.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,11 @@
2828

2929
from pyiceberg.partitioning import PartitionField, PartitionSpec
3030
from pyiceberg.schema import Schema
31-
from pyiceberg.transforms import IdentityTransform
31+
from pyiceberg.transforms import IdentityTransform, DayTransform
3232
from pyiceberg.types import (
3333
IntegerType,
3434
NestedField,
35+
TimestampType,
3536
)
3637

3738

@@ -572,23 +573,23 @@ def avro_schema_manifest_entry() -> Dict[str, Any]:
572573
def test_schema() -> Schema:
573574
return Schema(
574575
NestedField(1, "VendorID", IntegerType(), False),
575-
NestedField(2, "tpep_pickup_datetime", IntegerType(), False),
576+
NestedField(2, "tpep_pickup_datetime", TimestampType(), False),
576577
)
577578

578579

579580
@pytest.fixture(scope="session")
580581
def test_partition_spec() -> Schema:
581582
return PartitionSpec(
582583
PartitionField(1, 1000, IdentityTransform(), "VendorID"),
583-
PartitionField(2, 1001, IdentityTransform(), "tpep_pickup_datetime"),
584+
PartitionField(2, 1001, DayTransform(), "tpep_pickup_datetime"),
584585
)
585586

586587

587588
@pytest.fixture(scope="session")
588589
def generated_manifest_entry_file(
589-
avro_schema_manifest_entry: Dict[str, Any],
590-
test_schema: Schema,
591-
test_partition_spec: PartitionSpec,
590+
avro_schema_manifest_entry: Dict[str, Any],
591+
test_schema: Schema,
592+
test_partition_spec: PartitionSpec,
592593
) -> Generator[str, None, None]:
593594
from fastavro import parse_schema, writer
594595

0 commit comments

Comments
 (0)