Skip to content

Commit f8cf4b4

Browse files
committed
fix value_count test
1 parent 56e76d9 commit f8cf4b4

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

pandas/tests/extension/base/methods.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,10 @@ def test_value_counts(self, all_data, dropna):
3838

3939
def test_value_counts_with_normalize(self, data):
4040
# GH 33172
41-
data = data[:10].unique()
41+
data = data[:10].unique().remove_unused_categories()
4242
values = np.array(data[~data.isna()])
43-
ser = pd.Series(data, dtype=data.dtype)
4443

45-
result = ser.value_counts(normalize=True).sort_index()
46-
result = result[result > 0]
44+
result = pd.Series(data).value_counts(normalize=True).sort_index()
4745

4846
expected = pd.Series([1 / len(values)] * len(values), index=result.index)
4947
self.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)