Skip to content

Commit d71562e

Browse files
committed
simplify
1 parent d06b617 commit d71562e

File tree

1 file changed

+21
-17
lines changed

1 file changed

+21
-17
lines changed

test/lib/sources/huggingFaceSource.test.ts

Lines changed: 21 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -65,23 +65,27 @@ describe('parseHuggingFaceUrl', () => {
6565
})
6666

6767
test.for([
68-
['/datasets/namespace/repo', 'namespace/repo'],
69-
['/datasets/namespace/repo/', 'namespace/repo'],
70-
['/datasets/namespace/123', 'namespace/123'], /* all-number identifier is not a valid HF repo name, but we accept any string */
71-
])('returns a RepoURL for dataset repository URL: %s', ([path, repo]) => {
72-
const url = `https://huggingface.co${path}`
73-
expect(parseHuggingFaceUrl(url)).toEqual({
74-
kind: 'directory',
75-
origin,
76-
repo,
77-
source: url,
78-
action: 'tree',
79-
branch: 'main',
80-
path: '',
81-
})
82-
})
83-
84-
test.for([
68+
[
69+
'https://huggingface.co/datasets/namespace/repo',
70+
'https://huggingface.co/datasets/namespace/repo',
71+
'namespace/repo',
72+
'main',
73+
'',
74+
],
75+
[
76+
'https://huggingface.co/datasets/namespace/repo/',
77+
'https://huggingface.co/datasets/namespace/repo/',
78+
'namespace/repo',
79+
'main',
80+
'',
81+
],
82+
[
83+
'https://huggingface.co/datasets/namespace/123',
84+
'https://huggingface.co/datasets/namespace/123',
85+
// all-number identifier is not a valid HF repo name, but we accept any string
86+
'namespace/123',
87+
'main',
88+
''],
8589
[
8690
'https://huggingface.co/datasets/namespace/repo/tree/branch',
8791
'https://huggingface.co/datasets/namespace/repo/tree/branch',

0 commit comments

Comments
 (0)