Skip to content

Commit 9d82735

Browse files
committed
update: replace getAll -> get
1 parent ca4503e commit 9d82735

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

extensions/src/test/kotlin/com/fpf/smartscansdk/extensions/embeddings/FileEmbeddingStoreTest.kt

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ class FileEmbeddingStoreTest {
5151
)
5252

5353
store.add(embeddings)
54-
val loaded = store.getAll()
54+
val loaded = store.get()
5555

5656
Assertions.assertEquals(2, loaded.size)
5757
Assertions.assertEquals(embeddings[0].id, loaded[0].id)
@@ -72,7 +72,7 @@ class FileEmbeddingStoreTest {
7272
store.add(first)
7373
store.add(second)
7474

75-
val all = store.getAll()
75+
val all = store.get()
7676
Assertions.assertEquals(2, all.size)
7777
Assertions.assertEquals(1, all[0].id)
7878
Assertions.assertEquals(2, all[1].id)
@@ -90,7 +90,7 @@ class FileEmbeddingStoreTest {
9090
store.add(embeddings)
9191
store.remove(listOf(2L))
9292

93-
val remaining = store.getAll()
93+
val remaining = store.get()
9494
Assertions.assertEquals(2, remaining.size)
9595
Assertions.assertFalse(remaining.any { it.id == 2L })
9696
}
@@ -101,7 +101,7 @@ class FileEmbeddingStoreTest {
101101
val embeddings = listOf(embedding(1, 100, FloatArray(embeddingLength) { 0.1f }))
102102
store.add(embeddings)
103103

104-
val firstLoad = store.getAll()
104+
val firstLoad = store.get()
105105
if(store.useCache){
106106
Assertions.assertTrue(store.isCached)
107107
}else{
@@ -111,7 +111,7 @@ class FileEmbeddingStoreTest {
111111
store.clear()
112112
Assertions.assertFalse(store.isCached)
113113

114-
val secondLoad = store.getAll()
114+
val secondLoad = store.get()
115115
assertTrue(firstLoad.zip(secondLoad).all { (a, b) ->
116116
a.id == b.id && a.date == b.date && a.embeddings.contentEquals(b.embeddings)
117117
})
@@ -123,13 +123,13 @@ class FileEmbeddingStoreTest {
123123
val embeddings = listOf(embedding(1, 100, FloatArray(embeddingLength) { 0.1f }))
124124
store.add(embeddings)
125125

126-
val firstLoad = store.getAll()
126+
val firstLoad = store.get()
127127
Assertions.assertFalse(store.isCached)
128128

129129
store.clear()
130130
Assertions.assertFalse(store.isCached)
131131

132-
val secondLoad = store.getAll()
132+
val secondLoad = store.get()
133133
assertTrue(firstLoad.zip(secondLoad).all { (a, b) ->
134134
a.id == b.id && a.date == b.date && a.embeddings.contentEquals(b.embeddings)
135135
})

0 commit comments

Comments
 (0)