klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: linting (#414)

Browse Source
This commit is contained in:
cachho
2023-08-10 22:23:42 +02:00
committed by GitHub
parent 77e223be52
commit f0abfea55d
4 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/vectordb/test_chroma_db.py
View File

@@ -73,6 +73,7 @@ class TestChromaDbHostsLoglevel(unittest.TestCase):
self.assertEqual(mock_client.call_args[0][0].chroma_server_host, None)
self.assertEqual(mock_client.call_args[0][0].chroma_server_http_port, None)
class TestChromaDbDuplicateHandling:
def test_duplicates_throw_warning(self, caplog):
"""
@@ -101,8 +102,8 @@ class TestChromaDbDuplicateHandling:
app.collection.add(embeddings=[[0, 0, 0]], ids=["0"])
app.set_collection("test_collection_2")
app.collection.add(embeddings=[[0, 0, 0]], ids=["0"])
assert "Insert of existing embedding ID: 0" not in caplog.text # not
assert "Add of existing embedding ID: 0" not in caplog.text # not
assert "Insert of existing embedding ID: 0" not in caplog.text # not
assert "Add of existing embedding ID: 0" not in caplog.text # not
class TestChromaDbCollection(unittest.TestCase):
@@ -197,9 +198,9 @@ class TestChromaDbCollection(unittest.TestCase):
app2.collection.add(embeddings=[0, 0, 0], ids=["0"])
# Swap names and test
app1.set_collection('test_collection_2')
app1.set_collection("test_collection_2")
self.assertEqual(app1.count(), 1)
app2.set_collection('test_collection_1')
app2.set_collection("test_collection_1")
self.assertEqual(app2.count(), 3)
def test_ids_share_collections(self):