From cc0b446509fc07c48f4157c12189852fce62e817 Mon Sep 17 00:00:00 2001 From: Brian Hulette Date: Mon, 8 Aug 2022 13:33:23 -0700 Subject: [PATCH] Minor: Clean up an assertion in schemas_test (#22613) --- sdks/python/apache_beam/dataframe/schemas_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdks/python/apache_beam/dataframe/schemas_test.py b/sdks/python/apache_beam/dataframe/schemas_test.py index ed604bcd972c..42eea2bfbad4 100644 --- a/sdks/python/apache_beam/dataframe/schemas_test.py +++ b/sdks/python/apache_beam/dataframe/schemas_test.py @@ -164,7 +164,7 @@ def test_generate_proxy(self): 'max_speed': pd.Series(dtype=np.float64) }) - self.assertTrue(schemas.generate_proxy(Animal).equals(expected)) + pd.testing.assert_frame_equal(schemas.generate_proxy(Animal), expected) def test_nice_types_proxy_roundtrip(self): roundtripped = schemas.generate_proxy(