fix: doctest

i2mint · Jan 3, 2024 · 75fea8a · 75fea8a
1 parent ffadb09
commit 75fea8a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -93,8 +93,8 @@ You can read or write one document at a time too.
     >>> collection['cake'] = {
     ...     "documents": "contents for cake",
     ... }
-    >>> list(collection)
-    ['piece', 'of', 'cake']
+    >>> sorted(collection)  # sorting because order is not guaranteed
+    ['cake', 'of', 'piece']
     >>> assert collection['cake'] == {
     ...     'ids': ['cake'],
     ...     'embeddings': None,

diff --git a/chromadol/__init__.py b/chromadol/__init__.py
@@ -88,8 +88,8 @@
 >>> collection['cake'] = {
 ...     "documents": "contents for cake",
 ... }
->>> list(collection)
-['piece', 'of', 'cake']
+>>> sorted(collection)  # sorting because order is not guaranteed
+['cake', 'of', 'piece']
 >>> assert collection['cake'] == {
 ...     'ids': ['cake'],
 ...     'embeddings': None,

diff --git a/chromadol/tests/base_test.py b/chromadol/tests/base_test.py
@@ -61,7 +61,7 @@ def test_simple():
     collection['cake'] = {
         "documents": "contents for cake",
     }
-    assert list(collection) == ['piece', 'of', 'cake']
+    assert set(collection) == {'piece', 'of', 'cake'}
     assert collection['cake'] == {
         'ids': ['cake'],
         'embeddings': None,