Improve types for pd.options.display.* properties

Author: mike-flowers-airbnb

pandas-stubs/_config/config.pyi

@@ -49,30 +49,30 @@ class DisplayUnicode(DictWrapper):
 
 class Display(DictWrapper):
     chop_threshold: int | None
-    colheader_justify: str
+    colheader_justify: Literal["left", "right"]
     date_dayfirst: bool
     date_yearfirst: bool
     encoding: str
     expand_frame_repr: bool
     float_format: Callable[[float], str] | None
     html: DisplayHTML
-    large_repr: str
+    large_repr: Literal["truncate", "info"]
     latex: DisplayLaTeX
     max_categories: int
-    max_columns: int
-    max_colwidth: int
-    max_dir_items: int
+    max_columns: int | None
+    max_colwidth: int | None
+    max_dir_items: int | None
     max_info_columns: int
     max_info_rows: int
-    max_rows: int
-    max_seq_items: int
-    memory_usage: bool
-    min_rows: int
+    max_rows: int | None
+    max_seq_items: int | None
+    memory_usage: bool | Literal["deep"] | None
+    min_rows: int | None
     multi_sparse: bool
     notebook_repr_html: bool
     pprint_nest_depth: int
     precision: int
-    show_dimensions: str
+    show_dimensions: bool | Literal["truncate"]
     unicode: DisplayUnicode
     width: int
 

tests/test_config.py

@@ -48,3 +48,46 @@ def test_display_float_format():
     formatter = "{,.2f}".format
     with pd.option_context("display.float_format", formatter):
         assert pd.get_option("display.float_format") == formatter
+
+
+def test_display_types_none_allowed():
+    # GH 1230
+    properties_to_test = [
+        "max_columns",
+        "max_colwidth",
+        "max_dir_items",
+        "max_rows",
+        "max_seq_items",
+        "min_rows",
+    ]
+    for prop in properties_to_test:
+        # Test with None
+        setattr(pd.options.display, prop, None)
+        check(assert_type(pd.get_option(f"display.{prop}"), Any), type(None))
+
+        # Test with integer values
+        setattr(pd.options.display, prop, 100)
+        check(assert_type(pd.get_option(f"display.{prop}"), Any), int)
+
+
+def test_display_types_literal_constraints():
+    # GH 1230
+    # Various display options have specific allowed values, check assignment
+    # Test colheader_justify with allowed values
+    pd.options.display.colheader_justify = "left"
+    pd.options.display.colheader_justify = "right"
+
+    # Test large_repr with allowed values
+    pd.options.display.large_repr = "truncate"
+    pd.options.display.large_repr = "info"
+
+    # Test memory_usage with allowed values
+    pd.options.display.memory_usage = True
+    pd.options.display.memory_usage = False
+    pd.options.display.memory_usage = "deep"
+    pd.options.display.memory_usage = None
+
+    # Test show_dimensions with allowed values
+    pd.options.display.show_dimensions = True
+    pd.options.display.show_dimensions = False
+    pd.options.display.show_dimensions = "truncate"