There was an error while loading. Please reload this page.
1 parent 2284449 commit de14e5bCopy full SHA for de14e5b
1 file changed
tests/test_metastore.py
@@ -5,6 +5,7 @@
5
StructField,
6
StructType,
7
)
8
+from pyspark.testing import assertDataFrameEqual
9
from pytest import fixture, raises
10
11
from pysparkdt import reinit_local_metastore, spark_base
@@ -41,7 +42,7 @@ def test_reinit_requires_exactly_one_source():
41
42
def test_table_factories(spark: SparkSession):
43
reinit_local_metastore(spark, table_factories={TEST_TABLE: _build_table})
44
- rows = spark.read.format('delta').table(TEST_TABLE).collect()
45
- assert len(rows) == 2
46
- assert rows[0].id == 0
47
- assert rows[0].name == 'a'
+ actual = spark.read.format('delta').table(TEST_TABLE)
+ expected = _build_table(spark)
+ assertDataFrameEqual(actual, expected)
48
+
0 commit comments