From d394fbb1fccbe01df7d8dbc50495b3121a0c0ce9 Mon Sep 17 00:00:00 2001 From: Chalmer Lowe Date: Thu, 31 Aug 2023 14:02:21 +0000 Subject: [PATCH] bug: add import and object reference due to upstream changes --- tests/compliance/date/test_date_compliance.py | 3 ++- tests/compliance/time/test_time_compliance.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/compliance/date/test_date_compliance.py b/tests/compliance/date/test_date_compliance.py index 59c0968..5c43287 100644 --- a/tests/compliance/date/test_date_compliance.py +++ b/tests/compliance/date/test_date_compliance.py @@ -22,6 +22,7 @@ import pandas from pandas.tests.extension import base +import pandas._testing as tm import pytest import db_dtypes @@ -88,7 +89,7 @@ def test_value_counts(self, all_data, dropna): result = pandas.Series(all_data).value_counts(dropna=dropna).sort_index() expected = pandas.Series(other).value_counts(dropna=dropna).sort_index() - self.assert_series_equal(result, expected) + tm.assert_series_equal(result, expected) def test_diff(self): pytest.xfail( diff --git a/tests/compliance/time/test_time_compliance.py b/tests/compliance/time/test_time_compliance.py index ab1e050..3b36d16 100644 --- a/tests/compliance/time/test_time_compliance.py +++ b/tests/compliance/time/test_time_compliance.py @@ -22,6 +22,7 @@ import pandas from pandas.tests.extension import base +import pandas._testing as tm import pytest import db_dtypes @@ -88,7 +89,7 @@ def test_value_counts(self, all_data, dropna): result = pandas.Series(all_data).value_counts(dropna=dropna).sort_index() expected = pandas.Series(other).value_counts(dropna=dropna).sort_index() - self.assert_series_equal(result, expected) + tm.assert_series_equal(result, expected) class TestParsing(base.BaseParsingTests):