Skip to content
Snippets Groups Projects
Commit 7f57c8ff authored by Mingchung Xia's avatar Mingchung Xia
Browse files

Added memory usage logs

parent a8879270
No related branches found
No related tags found
1 merge request!13HNSW Implementation with Testcases
......@@ -20,6 +20,9 @@ final class HNSWTests: XCTestCase {
var corpus = HNSWCorpus(encoding: .glove6B50d)
corpus.addUntokenizedDocuments(docs)
let size = MemoryLayout.size(ofValue: corpus)
print("Approximate memory footprint: \(size) bytes")
XCTAssert(corpus.count == 3)
// Make sure none of our encodings are zero
......@@ -60,6 +63,8 @@ final class HNSWTests: XCTestCase {
var corpus = HNSWCorpus(encoding: .glove6B50d)
corpus.addUntokenizedDocuments(twentyQuotes)
let size = MemoryLayout.size(ofValue: corpus)
print("Approximate memory footprint: \(size) bytes")
XCTAssertEqual(corpus.count, 20)
......@@ -90,6 +95,10 @@ final class HNSWTests: XCTestCase {
corpus.addUntokenizedDocument(text)
}
}
let size = MemoryLayout.size(ofValue: corpus)
print("Approximate memory footprint: \(size) bytes")
//print("Loaded \(corpus.count) documents.")
XCTAssert(corpus.count == 17999)
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment