Fix tests

chromadb/test/test_api.py

@@ -91,6 +91,8 @@ def test_persist_index_loading(api_fixture, request):
     for key in nn.keys():
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
+        elif key == "included":
+            assert nn[key] == includes
         else:
             assert nn[key] is None
 
@@ -118,6 +120,8 @@ def __call__(self, input):
     for key in nn.keys():
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
+        elif key == "included":
+            assert nn[key] == includes
         else:
             assert nn[key] is None
 
@@ -146,6 +150,8 @@ def __call__(self, input):
     for key in nn.keys():
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
+        elif key == "included":
+            assert nn[key] == includes
         else:
             assert nn[key] is None
 
@@ -260,6 +266,8 @@ def test_get_from_db(api):
     for key in records.keys():
         if (key in includes) or (key == "ids"):
             assert len(records[key]) == 2
+        elif key == "included":
+            assert records[key] == includes
         else:
             assert records[key] is None
 
@@ -290,6 +298,8 @@ def test_get_nearest_neighbors(api):
     for key in nn.keys():
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
+        elif key == "included":
+            assert nn[key] == includes
         else:
             assert nn[key] is None
 
@@ -302,6 +312,8 @@ def test_get_nearest_neighbors(api):
     for key in nn.keys():
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
+        elif key == "included":
+            assert nn[key] == includes
         else:
             assert nn[key] is None
 
@@ -314,6 +326,8 @@ def test_get_nearest_neighbors(api):
     for key in nn.keys():
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 2
+        elif key == "included":
+            assert nn[key] == includes
         else:
             assert nn[key] is None
 
@@ -437,6 +451,8 @@ def test_increment_index_on(api):
     for key in nn.keys():
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
+        elif key == "included":
+            assert nn[key] == includes
         else:
             assert nn[key] is None
 
@@ -489,6 +505,8 @@ def test_peek(api):
     for key in peek.keys():
         if key in ["embeddings", "documents", "metadatas"] or key == "ids":
             assert len(peek[key]) == 2
+        elif key == "included":
+            assert peek[key] == ["embeddings", "metadatas", "documents"]
         else:
             assert peek[key] is None
 
@@ -1181,6 +1199,8 @@ def test_persist_index_loading_params(api, request):
     for key in nn.keys():
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
+        elif key == "included":
+            assert nn[key] == includes
         else:
             assert nn[key] is None
 
@@ -1299,6 +1319,8 @@ def test_get_nearest_neighbors_where_n_results_more_than_element(api):
     for key in results.keys():
         if key in includes or key == "ids":
             assert len(results[key][0]) == 2
+        elif key == "included":
+            assert results[key] == includes
         else:
             assert results[key] is None