Minor: Clean up an assertion in schemas_test (#22613)

apache · Aug 8, 2022 · cc0b446 · cc0b446
1 parent c9c161d
commit cc0b446
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(