updated value_counts subset test

tpackard1 · tpackard1 · commit e5f96b236589 · 2023-01-29T23:44:28.000-05:00
diff --git a/pandas/tests/frame/methods/test_value_counts.py b/pandas/tests/frame/methods/test_value_counts.py
@@ -160,13 +160,15 @@ def test_data_frame_value_counts_subset(nulls_fixture):
         index=pd.Index(["John", "Anne", "Beth"], name="first_name"),
     )
 
+    tm.assert_series_equal(result, expected)
+
     df = pd.DataFrame(
         {100: [2, 100, 5, 9], 200: [2, 6, 2, 6], 300: [4, 6, 2, 1]},
     )
     result = df.value_counts([200])
     expected = pd.Series(
         data=[2, 2],
-        index=pd.Index([2, 6], name=200),
+        index=pd.MultiIndex.from_arrays([[2, 6]], names=[200]),
     )
 
     tm.assert_series_equal(result, expected)

Original file line number	Diff line number	Diff line change
`@@ -160,13 +160,15 @@ def test_data_frame_value_counts_subset(nulls_fixture):`
`160`	`160`	`index=pd.Index(["John", "Anne", "Beth"], name="first_name"),`
`161`	`161`	`)`
`162`	`162`
	`163`	`+ tm.assert_series_equal(result, expected)`
	`164`	`+`
`163`	`165`	`df = pd.DataFrame(`
`164`	`166`	`{100: [2, 100, 5, 9], 200: [2, 6, 2, 6], 300: [4, 6, 2, 1]},`
`165`	`167`	`)`
`166`	`168`	`result = df.value_counts([200])`
`167`	`169`	`expected = pd.Series(`
`168`	`170`	`data=[2, 2],`
`169`		`- index=pd.Index([2, 6], name=200),`
	`171`	`+ index=pd.MultiIndex.from_arrays([[2, 6]], names=[200]),`
`170`	`172`	`)`
`171`	`173`
`172`	`174`	`tm.assert_series_equal(result, expected)`