Coverage

force_bdss/core_plugins/dummy/csv_extractor/tests/test_csv_extractor_data_source.py

 import unittest
 
 from force_bdss.core.data_value import DataValue
+from force_bdss.core.slot import Slot
 from force_bdss.core_plugins.dummy.csv_extractor.csv_extractor_data_source \
     import CSVExtractorDataSource
 from force_bdss.core_plugins.dummy.csv_extractor.csv_extractor_model import \
@@ -50,3 +51,12 @@ class TestCSVExtractorDataSource(unittest.TestCase):
         model.column = 50
         with self.assertRaises(IndexError):
             ds.run(model, mock_params)
+
+    def test_slots(self):
+        ds = CSVExtractorDataSource(self.bundle)
+        model = CSVExtractorModel(self.bundle)
+        slots = ds.slots(model)
+        self.assertEqual(len(slots), 2)
+        self.assertEqual(len(slots[0]), 0)
+        self.assertEqual(len(slots[1]), 1)
+        self.assertIsInstance(slots[1][0], Slot)

force_bdss/core_plugins/dummy/dummy_data_source/tests/test_dummy_data_source.py

 import unittest
 
+from force_bdss.core_plugins.dummy.dummy_data_source.dummy_data_source import \
+    DummyDataSource
+from force_bdss.core_plugins.dummy.dummy_data_source.dummy_data_source_model\
+    import \
+    DummyDataSourceModel
 from force_bdss.data_sources.base_data_source_bundle import \
     BaseDataSourceBundle
-from force_bdss.data_sources.tests.test_base_data_source import DummyDataSource
 
 try:
     import mock
@@ -17,3 +21,10 @@ class TestDummyDataSource(unittest.TestCase):
     def test_initialization(self):
         ds = DummyDataSource(self.bundle)
         self.assertEqual(ds.bundle, self.bundle)
+
+    def test_slots(self):
+        ds = DummyDataSource(self.bundle)
+        model = DummyDataSourceModel(self.bundle)
+        slots = ds.slots(model)
+        self.assertEqual(slots, ((), ()))
+