fix some index creation tests

marqo-ai · Sep 4, 2024 · 3ba4237 · 3ba4237
1 parent a9144d0
commit 3ba4237
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/v2_tests/test_create_index.py b/tests/v2_tests/test_create_index.py
@@ -38,10 +38,13 @@ def test_simple_index_creation(self):
         expected_settings = {
             'type': 'unstructured',
             'treatUrlsAndPointersAsImages': False,
+            'treatUrlsAndPointersAsMedia': False,
             'model': 'hf/e5-base-v2',
             'normalizeEmbeddings': True,
             'textPreprocessing': {'splitLength': 2, 'splitOverlap': 0, 'splitMethod': 'sentence'},
             'imagePreprocessing': {},
+            'audioPreprocessing': {'splitLength': 20, 'splitOverlap': 3},
+            'videoPreprocessing': {'splitLength': 20, 'splitOverlap': 3},
             'vectorNumericType': 'float',
             'filterStringMaxLength': 50,
             'annParameters': {
@@ -190,6 +193,8 @@ def test_create_simple_structured_index(self):
             'normalizeEmbeddings': True,
             'textPreprocessing': {'splitLength': 2, 'splitOverlap': 0, 'splitMethod': 'sentence'},
             'imagePreprocessing': {},
+            'audioPreprocessing': {'splitLength': 20, 'splitOverlap': 3},
+            'videoPreprocessing': {'splitLength': 20, 'splitOverlap': 3},
             'vectorNumericType': 'float',
             'annParameters': {'spaceType': 'prenormalized-angular', 'parameters': {'efConstruction': 512, 'm': 16}}}
         self.assertEqual(expected_index_settings, index_settings)