diff --git a/force_bdss/core_plugins/dummy/csv_extractor/tests/test_csv_extractor_data_source.py b/force_bdss/core_plugins/dummy/csv_extractor/tests/test_csv_extractor_data_source.py index a498b9e7e6ab847df6fc2c16ce47ce444c03eeb6..429b15ac1426a833dee4292429882c6a7adc9066 100644 --- a/force_bdss/core_plugins/dummy/csv_extractor/tests/test_csv_extractor_data_source.py +++ b/force_bdss/core_plugins/dummy/csv_extractor/tests/test_csv_extractor_data_source.py @@ -1,6 +1,7 @@ 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) diff --git a/force_bdss/core_plugins/dummy/dummy_data_source/tests/test_dummy_data_source.py b/force_bdss/core_plugins/dummy/dummy_data_source/tests/test_dummy_data_source.py index a77e99a6f1b2730222a73ad6d369beb071f8151b..ca6f2a3153b4226d014962713f0b1446055c5319 100644 --- a/force_bdss/core_plugins/dummy/dummy_data_source/tests/test_dummy_data_source.py +++ b/force_bdss/core_plugins/dummy/dummy_data_source/tests/test_dummy_data_source.py @@ -1,8 +1,12 @@ 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, ((), ())) +