File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change 14
14
import pandas .testing
15
15
import pytest
16
16
import sparse
17
+ from pandas .testing import assert_frame_equal
17
18
18
19
from ..catalog import in_memory
19
20
from ..catalog .adapter import CatalogContainerAdapter
20
21
from ..client import Context , from_context , record_history
21
- from ..client .dataframe import DataFrameClient
22
22
from ..mimetypes import PARQUET_MIMETYPE
23
23
from ..queries import Key
24
24
from ..server .app import build_app
@@ -520,7 +520,7 @@ def test_append_partition(
520
520
tree : CatalogContainerAdapter ,
521
521
orig_file : dict ,
522
522
file_toappend : dict ,
523
- expected_file : DataFrameClient ,
523
+ expected_file : dict ,
524
524
):
525
525
with Context .from_app (build_app (tree )) as context :
526
526
client = from_context (context , include_data_sources = True )
@@ -546,6 +546,4 @@ def test_append_partition(
546
546
547
547
df3 = pandas .DataFrame (expected_file )
548
548
549
- assert [row for col in x .columns for row in x [col ]] == [
550
- row for col in df3 .columns for row in df3 [col ]
551
- ]
549
+ assert_frame_equal (x .read (), df3 , check_dtype = False )
You can’t perform that action at this time.
0 commit comments