fix test failures

datafusion/tests/test_dataframe.py

@@ -415,7 +415,7 @@ def test_execution_plan(aggregate_df):
     assert expected == plan.display()
 
     # Check the number of partitions is as expected.
-    assert isinstance(type(plan.partition_count), int)
+    assert isinstance(plan.partition_count, int)
 
     expected = (
         "ProjectionExec: expr=[c1@0 as c1, SUM(test.c2)@1 as SUM(test.c2)]\n"
@@ -634,7 +634,7 @@ def test_empty_to_arrow_table(df):
 def test_to_pylist(df):
     # Convert datafusion dataframe to Python list
     pylist = df.to_pylist()
-    assert isinstance(type(pylist), list)
+    assert isinstance(pylist, list)
     assert pylist == [
         {"a": 1, "b": 4, "c": 8},
         {"a": 2, "b": 5, "c": 5},
@@ -645,7 +645,7 @@ def test_to_pylist(df):
 def test_to_pydict(df):
     # Convert datafusion dataframe to Python dictionary
     pydict = df.to_pydict()
-    assert isinstance(type(pydict), dict)
+    assert isinstance(pydict, dict)
     assert pydict == {"a": [1, 2, 3], "b": [4, 5, 6], "c": [8, 5, 8]}
 
 

datafusion/tests/test_substrait.py

@@ -40,7 +40,7 @@ def test_substrait_serialization(ctx):
     # For now just make sure the method calls blow up
     substrait_plan = ss.substrait.serde.serialize_to_plan("SELECT * FROM t", ctx)
     substrait_bytes = substrait_plan.encode()
-    assert isinstance(type(substrait_bytes), bytes)
+    assert isinstance(substrait_bytes, bytes)
     substrait_bytes = ss.substrait.serde.serialize_bytes("SELECT * FROM t", ctx)
     substrait_plan = ss.substrait.serde.deserialize_bytes(substrait_bytes)
     logical_plan = ss.substrait.consumer.from_substrait_plan(ctx, substrait_plan)