fixed pivot_table argument and test_pivot_table

Mariihmp · Mariihmp · commit affdb26b6b3e · 2024-12-24T23:50:12.000+03:30
diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi
@@ -1295,7 +1295,7 @@ class DataFrame(NDFrame, OpsMixin):
         dropna: _bool = ...,
         margins_name: _str = ...,
         observed: _bool = ...,
-        sort: _bool = True,
+        sort: _bool = ...,
     ) -> DataFrame: ...
     @overload
     def stack(
diff --git a/tests/test_frame.py b/tests/test_frame.py
@@ -1094,8 +1094,8 @@ def test_pivot_table_sort():
              'c': [5, 6],
              'd': [7, 8]}
         )
-        result = df.pivot_table(values='a', index='b', columns='c', sort=True)
-        assert result is not None
+        check(assert_type(df.pivot_table(values='a', index='b', columns='c', sort=True),
+                          pd.DataFrame), pd.DataFrame)
 
 
 def test_types_groupby_as_index() -> None:

Original file line number	Diff line number	Diff line change
`@@ -1094,8 +1094,8 @@ def test_pivot_table_sort():`
`1094`	`1094`	`'c': [5, 6],`
`1095`	`1095`	`'d': [7, 8]}`
`1096`	`1096`	`)`
`1097`		`- result = df.pivot_table(values='a', index='b', columns='c', sort=True)`
`1098`		`- assert result is not None`
	`1097`	`+ check(assert_type(df.pivot_table(values='a', index='b', columns='c', sort=True),`
	`1098`	`+ pd.DataFrame), pd.DataFrame)`
`1099`	`1099`
`1100`	`1100`
`1101`	`1101`	`def test_types_groupby_as_index() -> None:`