Skip to content

Commit b4e1c49

Browse files
committed
add test case for large file metadata
1 parent af26e60 commit b4e1c49

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

Tests/HubTests/HubApiTests.swift

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,25 @@ class HubApiTests: XCTestCase {
144144
XCTAssertGreaterThan(metadata.size!, 0)
145145
}
146146
}
147+
148+
func testGetLargeFileMetadata() async throws {
149+
do {
150+
let revision = "eaf97358a37d03fd48e5a87d15aff2e8423c1afb"
151+
let etag = "fc329090bfbb2570382c9af997cffd5f4b78b39b8aeca62076db69534e020107"
152+
let location = "https://cdn-lfs.hf.co/repos/4a/4e/4a4e587f66a2979dcd75e1d7324df8ee9ef74be3582a05bea31c2c26d0d467d0/fc329090bfbb2570382c9af997cffd5f4b78b39b8aeca62076db69534e020107?response-content-disposition=inline%3B+filename*%3DUTF-8%27%27model.mlmodel%3B+filename%3D%22model.mlmodel"
153+
let size = 504766
154+
155+
let url = URL(string: "https://huggingface.co/coreml-projects/Llama-2-7b-chat-coreml/resolve/main/llama-2-7b-chat.mlpackage/Data/com.apple.CoreML/model.mlmodel")
156+
let metadata = try await Hub.getFileMetadata(fileURL: url!)
157+
158+
XCTAssertEqual(metadata.commitHash, revision)
159+
XCTAssertNotNil(metadata.etag, etag)
160+
XCTAssertTrue(metadata.location.contains(location))
161+
XCTAssertEqual(metadata.size, size)
162+
} catch {
163+
XCTFail("\(error)")
164+
}
165+
}
147166
}
148167

149168
class SnapshotDownloadTests: XCTestCase {

0 commit comments

Comments
 (0)