class KascadeLdfSimulationTest(unittest.TestCase):
    def setUp(self):
        cluster = Mock()
        data = MagicMock()
        output = "/simulations"
        R = sentinel.R
        N = MagicMock()

        self.simulation = KascadeLdfSimulation(cluster, data, output, R, N)

        # make progressbar(list) do nothing (i.e., return list)
        self.progressbar_patcher = patch("progressbar.ProgressBar")
        self.progressbar_mock = self.progressbar_patcher.start()
        self.progressbar_mock.return_value.side_effect = lambda x: x

    def test_init_calls_super_init_with_all_args(self):
        with patch.object(BaseLdfSimulation, "__init__") as mock_base_init:
            cluster = sentinel.cluster
            data = sentinel.data
            output = sentinel.output
            R = sentinel.R
            N = sentinel.N
            sim = KascadeLdfSimulation(cluster, data, output, R, N=N)
            mock_base_init.assert_called_once_with(cluster, data, output, R, N=N)

    def test_calculate_ldf_value_useful_immediately_after_init(self):
        self.simulation.calculate_ldf_value(1.0, 10 ** 4.8)

    def test_run_calls_super_run(self):
        with patch.object(BaseLdfSimulation, "run") as mock_run:
            self.simulation.run()
            mock_run.assert_called_once_with()

    def test_calculate_ldf_value(self):
        r = logspace(1, 3, 10)
        expected = array(
            [
                1.39434005e01,
                7.49964318e00,
                3.49327708e00,
                1.37707188e00,
                4.57377254e-01,
                1.30067636e-01,
                3.26255672e-02,
                7.46150200e-03,
                1.60164385e-03,
                3.29852843e-04,
            ]
        )
        actual = self.simulation.calculate_ldf_value(r, 10 ** 4.8)

        self.assertTrue(((expected - actual) / expected < 1e-8).all())
    def setUp(self):
        cluster = Mock()
        data = MagicMock()
        output = '/simulations'
        R = sentinel.R
        N = MagicMock()

        self.simulation = KascadeLdfSimulation(cluster, data, output, R, N)

        # make progressbar(list) do nothing (i.e., return list)
        self.progressbar_patcher = patch('progressbar.ProgressBar')
        self.progressbar_mock = self.progressbar_patcher.start()
        self.progressbar_mock.return_value.side_effect = lambda x: x
 def test_init_calls_super_init_with_all_args(self):
     with patch.object(BaseLdfSimulation, '__init__') as mock_base_init:
         cluster = sentinel.cluster
         data = sentinel.data
         output = sentinel.output
         R = sentinel.R
         N = sentinel.N
         sim = KascadeLdfSimulation(cluster, data, output, R, N=N)
         mock_base_init.assert_called_once_with(cluster, data, output, R, N=N)
class KascadeLdfSimulationTest(unittest.TestCase):
    def setUp(self):
        cluster = Mock()
        data = MagicMock()
        output = '/simulations'
        R = sentinel.R
        N = MagicMock()

        self.simulation = KascadeLdfSimulation(cluster, data, output, R, N)

        # make progressbar(list) do nothing (i.e., return list)
        self.progressbar_patcher = patch('progressbar.ProgressBar')
        self.progressbar_mock = self.progressbar_patcher.start()
        self.progressbar_mock.return_value.side_effect = lambda x: x

    def test_init_calls_super_init_with_all_args(self):
        with patch.object(BaseLdfSimulation, '__init__') as mock_base_init:
            cluster = sentinel.cluster
            data = sentinel.data
            output = sentinel.output
            R = sentinel.R
            N = sentinel.N
            sim = KascadeLdfSimulation(cluster, data, output, R, N=N)
            mock_base_init.assert_called_once_with(cluster, data, output, R, N=N)

    def test_calculate_ldf_value_useful_immediately_after_init(self):
        self.simulation.calculate_ldf_value(1., 10 ** 4.8)

    def test_run_calls_super_run(self):
        with patch.object(BaseLdfSimulation, 'run') as mock_run:
            self.simulation.run()
            mock_run.assert_called_once_with()

    def test_calculate_ldf_value(self):
        r = logspace(1, 3, 10)
        expected = array([  1.39434005e+01, 7.49964318e+00, 3.49327708e+00,
                            1.37707188e+00, 4.57377254e-01, 1.30067636e-01,
                            3.26255672e-02, 7.46150200e-03, 1.60164385e-03,
                            3.29852843e-04])
        actual = self.simulation.calculate_ldf_value(r, 10 ** 4.8)

        self.assertTrue(((expected - actual) / expected < 1e-8).all())
    def setUp(self):
        cluster = Mock()
        data = MagicMock()
        output = "/simulations"
        R = sentinel.R
        N = MagicMock()

        self.simulation = KascadeLdfSimulation(cluster, data, output, R, N)

        # make progressbar(list) do nothing (i.e., return list)
        self.progressbar_patcher = patch("progressbar.ProgressBar")
        self.progressbar_mock = self.progressbar_patcher.start()
        self.progressbar_mock.return_value.side_effect = lambda x: x