Skip to content

GitLab

Unverified Commit 4b295d60 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[FIX] fix compute/test_data.py::test_csvdataset (#3643)

parent 6dce19d8
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/compute/test_data.py
View file @ 4b295d60
......@@ -716,7 +716,7 @@ def _test_DGLCSVDataset_single():
# remove original node data file to verify reload from cached files
os.remove(nodes_csv_path_0)
assert not os.path.exists(nodes_csv_path_0)
csv_dataset = data.DGLCSVDataset(
csv_dataset = csv_ds.DGLCSVDataset(
test_dir, force_reload=force_reload)
assert len(csv_dataset) == 1
g = csv_dataset[0]
......@@ -799,7 +799,7 @@ def _test_DGLCSVDataset_multiple():
# remove original node data file to verify reload from cached files
os.remove(nodes_csv_path_0)
assert not os.path.exists(nodes_csv_path_0)
csv_dataset = data.DGLCSVDataset(
csv_dataset = csv_ds.DGLCSVDataset(
test_dir, force_reload=force_reload)
assert len(csv_dataset) == num_graphs
assert csv_dataset.has_cache()
......@@ -885,7 +885,7 @@ def _test_DGLCSVDataset_customized_data_parser():
data[header] = dt
return data
# load CSVDataset with customized node/edge/graph_data_parser
csv_dataset = data.DGLCSVDataset(
csv_dataset = csv_ds.DGLCSVDataset(
test_dir, node_data_parser={'user': CustDataParser()}, edge_data_parser={('user', 'like', 'item'): CustDataParser()}, graph_data_parser=CustDataParser())
assert len(csv_dataset) == num_graphs
assert len(csv_dataset.data) == 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment