add test to check if nullability is maintained

delta-io · Nov 17, 2023 · c9cf446 · c9cf446
1 parent ad5b2c3
commit c9cf446
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/python/tests/test_schema.py b/python/tests/test_schema.py
@@ -230,6 +230,36 @@ def test_delta_schema():
             pa.schema([("some_int", pa.int32()), ("some_string", pa.string())]),
             False,
         ),
+        (
+            pa.schema(
+                [
+                    pa.field("some_int", pa.uint32(), nullable=True),
+                    pa.field("some_string", pa.string(), nullable=False),
+                ]
+            ),
+            pa.schema(
+                [
+                    pa.field("some_int", pa.int32(), nullable=True),
+                    pa.field("some_string", pa.string(), nullable=False),
+                ]
+            ),
+            False,
+        ),
+        (
+            pa.schema(
+                [
+                    pa.field("some_int", pa.uint32(), nullable=True),
+                    pa.field("some_string", pa.string(), nullable=False),
+                ]
+            ),
+            pa.schema(
+                [
+                    pa.field("some_int", pa.int32(), nullable=True),
+                    pa.field("some_string", pa.large_string(), nullable=False),
+                ]
+            ),
+            True,
+        ),
         (
             pa.schema([("some_int", pa.uint32()), ("some_string", pa.string())]),
             pa.schema([("some_int", pa.int32()), ("some_string", pa.large_string())]),