Skip to content

Commit

Permalink
Remove redundant prefix.
Browse files Browse the repository at this point in the history
  • Loading branch information
clarkzinzow committed Feb 21, 2024
1 parent b0b3de5 commit 4f9b26d
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 9 deletions.
6 changes: 3 additions & 3 deletions tests/io/delta_lake/test_table_read.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@


@pytest.mark.integration()
def test_daft_deltalake_read_basic(tmp_path):
def test_deltalake_read_basic(tmp_path):
pd_df = pd.DataFrame(
{
"a": [1, 2, 3],
Expand All @@ -28,15 +28,15 @@ def test_daft_deltalake_read_basic(tmp_path):


@pytest.mark.integration()
def test_daft_deltalake_read_full(local_deltalake_table):
def test_deltalake_read_full(local_deltalake_table):
path, dfs = local_deltalake_table
df = daft.read_delta_lake(str(path))
assert df.schema() == Schema.from_pyarrow_schema(deltalake.DeltaTable(path).schema().to_pyarrow())
pd.testing.assert_frame_equal(df.to_pandas(), pd.concat(dfs).reset_index(drop=True))


@pytest.mark.integration()
def test_daft_deltalake_read_show(local_deltalake_table):
def test_deltalake_read_show(local_deltalake_table):
path, _ = local_deltalake_table
df = daft.read_delta_lake(str(path))
df.show()
12 changes: 6 additions & 6 deletions tests/io/delta_lake/test_table_read_pushdowns.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@


@pytest.mark.integration()
def test_daft_deltalake_read_predicate_pushdown_on_data(local_deltalake_table):
def test_deltalake_read_predicate_pushdown_on_data(local_deltalake_table):
path, dfs = local_deltalake_table
df = daft.read_delta_lake(str(path))
df = df.where(df["a"] == 2)
Expand All @@ -23,7 +23,7 @@ def test_daft_deltalake_read_predicate_pushdown_on_data(local_deltalake_table):


@pytest.mark.integration()
def test_daft_deltalake_read_predicate_pushdown_on_part(local_deltalake_table):
def test_deltalake_read_predicate_pushdown_on_part(local_deltalake_table):
path, dfs = local_deltalake_table
df = daft.read_delta_lake(str(path))
df = df.where(df["part_idx"] == 2)
Expand All @@ -34,7 +34,7 @@ def test_daft_deltalake_read_predicate_pushdown_on_part(local_deltalake_table):


@pytest.mark.integration()
def test_daft_deltalake_read_predicate_pushdown_on_part_non_eq(local_deltalake_table):
def test_deltalake_read_predicate_pushdown_on_part_non_eq(local_deltalake_table):
path, dfs = local_deltalake_table
df = daft.read_delta_lake(str(path))
df = df.where(df["part_idx"] < 3)
Expand All @@ -45,7 +45,7 @@ def test_daft_deltalake_read_predicate_pushdown_on_part_non_eq(local_deltalake_t


@pytest.mark.integration()
def test_daft_deltalake_read_predicate_pushdown_on_part_and_data(local_deltalake_table):
def test_deltalake_read_predicate_pushdown_on_part_and_data(local_deltalake_table):
path, dfs = local_deltalake_table
df = daft.read_delta_lake(str(path))
df = df.where((df["part_idx"] == 2) & (df["e"] == datetime.datetime(2024, 2, 11)))
Expand All @@ -59,7 +59,7 @@ def test_daft_deltalake_read_predicate_pushdown_on_part_and_data(local_deltalake


@pytest.mark.integration()
def test_daft_deltalake_read_predicate_pushdown_on_part_and_data_same_clause(local_deltalake_table):
def test_deltalake_read_predicate_pushdown_on_part_and_data_same_clause(local_deltalake_table):
path, dfs = local_deltalake_table
df = daft.read_delta_lake(str(path))
df = df.where(df["part_idx"] < df["a"])
Expand All @@ -71,7 +71,7 @@ def test_daft_deltalake_read_predicate_pushdown_on_part_and_data_same_clause(loc


@pytest.mark.integration()
def test_daft_deltalake_read_predicate_pushdown_on_part_empty(local_deltalake_table):
def test_deltalake_read_predicate_pushdown_on_part_empty(local_deltalake_table):
path, dfs = local_deltalake_table
df = daft.read_delta_lake(str(path))
# There should only be len(dfs) - 1 partitions; see local_deltalake_table fixture.
Expand Down

0 comments on commit 4f9b26d

Please sign in to comment.