From 7f57c8ff83c345c18ca7c8d27e9b77f70f6fb270 Mon Sep 17 00:00:00 2001
From: Mingchung Xia <mingchung.xia@gmail.com>
Date: Wed, 7 Feb 2024 23:08:48 -0500
Subject: [PATCH] Added memory usage logs
---
Tests/SwiftNLPTests/2. Encoding/HNSWTests.swift | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/Tests/SwiftNLPTests/2. Encoding/HNSWTests.swift b/Tests/SwiftNLPTests/2. Encoding/HNSWTests.swift
index 844fff6c..fd8fc7c7 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)
}
--
GitLab