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)