Create database index for samples

predictive_punter/provider.py

@@ -6,6 +6,17 @@
 class Provider(racing_data.Provider):
     """Extend the racing_data Provider class with additional functionality specific to predictive analytics"""
 
+    @property
+    def database_indexes(self):
+        """Return a dictionary of required database indexes for each entity type"""
+
+        database_indexes = super().database_indexes
+        database_indexes[Sample] = [
+            [('runner_id', 1)]
+        ]
+
+        return database_indexes
+
     def get_runner_by_sample(self, sample):
         """Get the runner associated with the specified sample"""
 

tests/provider/indexes_test.py

@@ -0,0 +1,17 @@
+import predictive_punter
+
+
+def test_indexes(provider):
+    """The provider should create all necessary database indexes during initialisation"""
+
+    expected_indexes = {
+        predictive_punter.Sample:   [
+            [('runner_id', 1)]
+        ]
+    }
+
+    for entity_type in expected_indexes:
+        collection = provider.get_database_collection(entity_type)
+        index_keys = [index['key'] for index in collection.index_information().values()]
+        for expected_index in expected_indexes[entity_type]:
+            assert expected_index in index_keys