diff --git a/Tests/SwiftNLPTests/2. Encoding/HNSWTests.swift b/Tests/SwiftNLPTests/2. Encoding/HNSWTests.swift index 844fff6c3e1b27581c30ceace0c3b7a955d916f1..fd8fc7c75c65b8ef80e76208034c562adff43716 100644 --- a/Tests/SwiftNLPTests/2. Encoding/HNSWTests.swift +++ b/Tests/SwiftNLPTests/2. Encoding/HNSWTests.swift @@ -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) }