From 0123e2af8d564f9299ed832df834c78c86fb3b2c Mon Sep 17 00:00:00 2001 From: Stefano Borini <sborini@enthought.com> Date: Tue, 1 Aug 2017 14:08:08 +0100 Subject: [PATCH] Coverage --- .../tests/test_kpi_adder_calculator.py | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 force_bdss/core_plugins/dummy/kpi_adder/tests/test_kpi_adder_calculator.py diff --git a/force_bdss/core_plugins/dummy/kpi_adder/tests/test_kpi_adder_calculator.py b/force_bdss/core_plugins/dummy/kpi_adder/tests/test_kpi_adder_calculator.py new file mode 100644 index 0000000..6f325ef --- /dev/null +++ b/force_bdss/core_plugins/dummy/kpi_adder/tests/test_kpi_adder_calculator.py @@ -0,0 +1,27 @@ +import unittest + +from force_bdss.core.data_value import DataValue +from force_bdss.core_plugins.dummy.kpi_adder.kpi_adder_model import \ + KPIAdderModel +from force_bdss.kpi.base_kpi_calculator_bundle import BaseKPICalculatorBundle + +try: + import mock +except ImportError: + from unittest import mock + +from force_bdss.core_plugins.dummy.kpi_adder.kpi_adder_calculator import \ + KPIAdderCalculator + + +class TestKPIAdderCalculator(unittest.TestCase): + def test_basic_functionality(self): + kpic = KPIAdderCalculator(mock.Mock(spec=BaseKPICalculatorBundle)) + model = KPIAdderModel(mock.Mock(spec=BaseKPICalculatorBundle)) + model.cuba_type_in = "PRESSURE" + model.cuba_type_out = "TOTAL_PRESSURE" + dv1 = DataValue(type="PRESSURE", value=10) + dv2 = DataValue(type="PRESSURE", value=30) + res = kpic.run(model, [dv1, dv2]) + self.assertEqual(res[0].type, "TOTAL_PRESSURE") + self.assertEqual(res[0].value, 40) -- GitLab