🐛 Persist index uniqueness (#801)

* 🔨 Add "unique" key to be saved * ✅ Update unit tests * Update schema_inference.rst Co-authored-by: Niels Bantilan <niels.bantilan@gmail.com>
unionai-oss · Apr 1, 2022 · c0c2d40 · c0c2d40
1 parent 20fd29a
commit c0c2d40
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 0 deletions.
diff --git a/docs/source/schema_inference.rst b/docs/source/schema_inference.rst
@@ -216,6 +216,7 @@ is a convenience method for this functionality.
         greater_than_or_equal_to: 0.0
         less_than_or_equal_to: 2.0
       name: null
+      unique: false
       coerce: false
     coerce: true
     strict: false

diff --git a/pandera/schema_statistics.py b/pandera/schema_statistics.py
@@ -114,6 +114,7 @@ def _get_series_base_schema_statistics(series_schema_base):
         "checks": parse_checks(series_schema_base.checks),
         "coerce": series_schema_base.coerce,
         "name": series_schema_base.name,
+        "unique": series_schema_base.unique,
     }
 
 

diff --git a/tests/core/test_schema_statistics.py b/tests/core/test_schema_statistics.py
@@ -475,6 +475,7 @@ def test_get_dataframe_schema_statistics():
                 "nullable": False,
                 "coerce": False,
                 "name": "int_index",
+                "unique": False,
             }
         ],
         "coerce": False,
@@ -503,6 +504,7 @@ def test_get_series_schema_statistics():
         },
         "name": None,
         "coerce": False,
+        "unique": False,
     }
 
 
@@ -529,6 +531,7 @@ def test_get_series_schema_statistics():
                     },
                     "name": "int_index",
                     "coerce": False,
+                    "unique": False,
                 }
             ],
         ]

diff --git a/tests/io/test_io.py b/tests/io/test_io.py
@@ -213,6 +213,7 @@ def _create_schema(index="single"):
   nullable: false
   checks: null
   name: null
+  unique: false
   coerce: false
 coerce: false
 strict: true