def test_further_elements_already_buffered(self): time = pd.date_range('2000-01-01', freq='24H', periods=7) step = Step(self.module_mock, {"x": self.step_mock}, file_manager=MagicMock()) step.buffer = {"STEP": xr.DataArray([2, 3, 4, 3, 3, 1, 2], dims=["time"], coords={'time': time})} result = step.further_elements(pd.Timestamp("2000-01-05")) self.step_mock.further_elements.assert_not_called() self.assertEqual(result, True)
def test_further_elements_target_false(self): target_step = MagicMock() target_step.further_elements.return_value = False time = pd.date_range('2000-01-01', freq='1H', periods=7) step = Step(self.module_mock, {"x": self.step_mock}, targets={"target": target_step}, file_manager=MagicMock()) step.buffer = {"STEP": xr.DataArray([2, 3, 4, 3, 3, 1, 2], dims=["time"], coords={'time': time})} result = step.further_elements(pd.Timestamp("2000.12.12")) target_step.further_elements.assert_called_once_with(pd.Timestamp("2000.12.12")) self.assertFalse(result)