Increase coverage

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

+import unittest
+
+from force_bdss.core_plugins.dummy.tests.data_source_bundle_test_mixin import \
+    DataSourceBundleTestMixin
+from force_bdss.core_plugins.dummy.csv_extractor.csv_extractor_bundle import \
+    CSVExtractorBundle
+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 \
+    CSVExtractorModel
+
+
+class TestCSVExtractorBundle(DataSourceBundleTestMixin, unittest.TestCase):
+    @property
+    def bundle_class(self):
+        return CSVExtractorBundle
+
+    @property
+    def model_class(self):
+        return CSVExtractorModel
+
+    @property
+    def data_source_class(self):
+        return CSVExtractorDataSource

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

+import unittest
+
+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 \
+    CSVExtractorModel
+from force_bdss.data_sources.base_data_source_bundle import \
+    BaseDataSourceBundle
+from force_bdss.data_sources.data_source_result import DataSourceResult
+from force_bdss.tests import fixtures
+
+try:
+    import mock
+except ImportError:
+    from unittest import mock
+
+
+class TestCSVExtractorDataSource(unittest.TestCase):
+    def setUp(self):
+        self.bundle = mock.Mock(spec=BaseDataSourceBundle)
+
+    def test_initialization(self):
+        ds = CSVExtractorDataSource(self.bundle)
+        self.assertEqual(ds.bundle, self.bundle)
+
+    def test_run(self):
+        ds = CSVExtractorDataSource(self.bundle)
+        model = CSVExtractorModel(self.bundle)
+        model.filename = fixtures.get("foo.csv")
+        mock_params = mock.Mock()
+        mock_params.values = [1.0]
+        result = ds.run(model, mock_params)
+        self.assertIsInstance(result, DataSourceResult)

force_bdss/core_plugins/dummy/dummy_dakota/tests/test_dakota_bundle.py

+import unittest
+
+from envisage.plugin import Plugin
+
+from force_bdss.core_plugins.dummy.dummy_dakota.dakota_bundle import \
+    DummyDakotaBundle
+from force_bdss.core_plugins.dummy.dummy_dakota.dakota_model import \
+    DummyDakotaModel
+from force_bdss.core_plugins.dummy.dummy_dakota.dakota_optimizer import \
+    DummyDakotaOptimizer
+
+try:
+    import mock
+except ImportError:
+    from unittest import mock
+
+
+class TestDakotaBundle(unittest.TestCase):
+    def setUp(self):
+        self.plugin = mock.Mock(spec=Plugin)
+
+    def test_initialization(self):
+        bundle = DummyDakotaBundle(self.plugin)
+        self.assertIn("dummy_dakota", bundle.id)
+        self.assertEqual(bundle.plugin, self.plugin)
+
+    def test_create_model(self):
+        bundle = DummyDakotaBundle(self.plugin)
+        model = bundle.create_model({})
+        self.assertIsInstance(model, DummyDakotaModel)
+
+        model = bundle.create_model()
+        self.assertIsInstance(model, DummyDakotaModel)
+
+    def test_create_mco(self):
+        bundle = DummyDakotaBundle(self.plugin)
+        ds = bundle.create_optimizer()
+        self.assertIsInstance(ds, DummyDakotaOptimizer)
+
+    def test_create_communicator(self):
+        bundle = DummyDakotaBundle(self.plugin)
+        ds = bundle.create_optimizer()
+        self.assertIsInstance(ds, DummyDakotaOptimizer)

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

+import unittest
+
+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
+except ImportError:
+    from unittest import mock
+
+
+class TestDummyDataSource(unittest.TestCase):
+    def setUp(self):
+        self.bundle = mock.Mock(spec=BaseDataSourceBundle)
+
+    def test_initialization(self):
+        ds = DummyDataSource(self.bundle)
+        self.assertEqual(ds.bundle, self.bundle)

force_bdss/core_plugins/dummy/dummy_data_source/tests/test_dummy_data_source_bundle.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_bundle import DummyDataSourceBundle
+from force_bdss.core_plugins.dummy.dummy_data_source.dummy_data_source_model\
+    import DummyDataSourceModel
+from force_bdss.core_plugins.dummy.tests.data_source_bundle_test_mixin import \
+    DataSourceBundleTestMixin
+
+
+class TestDummyDataSourceBundle(DataSourceBundleTestMixin, unittest.TestCase):
+    @property
+    def bundle_class(self):
+        return DummyDataSourceBundle
+
+    @property
+    def model_class(self):
+        return DummyDataSourceModel
+
+    @property
+    def data_source_class(self):
+        return DummyDataSource

force_bdss/core_plugins/dummy/tests/data_source_bundle_test_mixin.py

+from envisage.plugin import Plugin
+
+try:
+    import mock
+except ImportError:
+    from unittest import mock
+
+
+class DataSourceBundleTestMixin(object):
+    def setUp(self):
+        self.plugin = mock.Mock(spec=Plugin)
+        super(DataSourceBundleTestMixin, self).setUp()
+
+    # Note: we can't use metaclasses. Apparently using six.with_metaclass
+    # breaks the unittest TestCase mechanism. py3 metaclassing works.
+    @property
+    def bundle_class(self):
+        raise NotImplementedError()
+
+    @property
+    def model_class(self):
+        raise NotImplementedError()
+
+    @property
+    def data_source_class(self):
+        raise NotImplementedError()
+
+    def test_initialization(self):
+        bundle = self.bundle_class(self.plugin)
+        self.assertNotEqual(bundle.id, "")
+        self.assertEqual(bundle.plugin, self.plugin)
+
+    def test_create_model(self):
+        bundle = self.bundle_class(self.plugin)
+        model = bundle.create_model({})
+        self.assertIsInstance(model, self.model_class)
+
+        model = bundle.create_model()
+        self.assertIsInstance(model, self.model_class)
+
+    def test_create_data_source(self):
+        bundle = self.bundle_class(self.plugin)
+        ds = bundle.create_data_source()
+        self.assertIsInstance(ds, self.data_source_class)