|
5 | 5 | import numpy as np
|
6 | 6 | import pytest
|
7 | 7 |
|
| 8 | +from pandas._config.config import option_context |
| 9 | + |
8 | 10 | import pandas.util._test_decorators as td
|
9 | 11 | from pandas.util._test_decorators import (
|
10 | 12 | async_mark,
|
@@ -87,6 +89,25 @@ def test_tab_completion(self):
|
87 | 89 | assert key not in dir(df)
|
88 | 90 | assert isinstance(df.__getitem__("A"), DataFrame)
|
89 | 91 |
|
| 92 | + def test_display_max_dir_items(self): |
| 93 | + # display.max_dir_items increaes the number of columns that are in __dir__. |
| 94 | + columns = ["a" + str(i) for i in range(420)] |
| 95 | + values = [range(420), range(420)] |
| 96 | + df = DataFrame(values, columns=columns) |
| 97 | + |
| 98 | + # The default value for display.max_dir_items is 100 |
| 99 | + assert "a99" in dir(df) |
| 100 | + assert "a100" not in dir(df) |
| 101 | + |
| 102 | + with option_context("display.max_dir_items", 300): |
| 103 | + df = DataFrame(values, columns=columns) |
| 104 | + assert "a299" in dir(df) |
| 105 | + assert "a300" not in dir(df) |
| 106 | + |
| 107 | + with option_context("display.max_dir_items", None): |
| 108 | + df = DataFrame(values, columns=columns) |
| 109 | + assert "a419" in dir(df) |
| 110 | + |
90 | 111 | def test_not_hashable(self):
|
91 | 112 | empty_frame = DataFrame()
|
92 | 113 |
|
|
0 commit comments