Skip to content

Commit 1786607

Browse files
committed
update dataset viewer workflow test
1 parent 47c64fe commit 1786607

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/api/dataset_viewer_test.go

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,9 +75,9 @@ configs:
7575
- config_name: defaultgo
7676
data_files:
7777
- split: traingo
78-
path: "train/0.parquet"
78+
path: "train/*.parquet"
7979
- split: testgo
80-
path: "test/1.parquet"
80+
path: "test/*.parquet"
8181
---
8282
`
8383
newContent := configContent + fileContent
@@ -96,13 +96,15 @@ configs:
9696
require.NoError(t, err)
9797
err = gitCommitAndPush(dir)
9898
require.NoError(t, err)
99+
// wait temporal workflow
100+
time.Sleep(5 * time.Second)
99101

100102
resp, err = userClient.Get("http://localhost:9091/api/v1/datasets/user/test/dataviewer/catalog")
101103
require.NoError(t, err)
102104
defer resp.Body.Close()
103105
body, err = io.ReadAll(resp.Body)
104106
require.NoError(t, err)
105-
expected := `{"msg":"OK","data":{"configs":[{"config_name":"defaultgo","data_files":[{"split":"traingo","path":["train/0.parquet"]},{"split":"testgo","path":["test/1.parquet"]}]}],"dataset_info":[{"config_name":"defaultgo","splits":[{"name":"traingo","num_examples":20},{"name":"testgo","num_examples":20}]}],"status":0,"logs":""}}
107+
expected := `{"msg":"OK","data":{"configs":[{"config_name":"defaultgo","data_files":[{"split":"traingo","path":["train/*.parquet"]},{"split":"testgo","path":["test/*.parquet"]}]}],"dataset_info":[{"config_name":"defaultgo","splits":[{"name":"traingo","num_examples":20},{"name":"testgo","num_examples":20}]}],"status":2,"logs":"Done"}}
106108
`
107109
require.Equal(t, expected, string(body))
108110

0 commit comments

Comments
 (0)