def test__sample_model_vine(self, qualified_mock):
"""_sample_model sample the number of rows from the given model."""
# Setup
data_navigator = MagicMock(spec=DataNavigator)
modeler = MagicMock(spec=Modeler)
sampler = Sampler(data_navigator, modeler)
model = MagicMock()
values = [
np.array([1, 1, 1]),
np.array([2, 2, 2]),
np.array([3, 3, 3])
]
qualified_mock.return_value = 'copulas.multivariate.vine.VineCopula'
model.sample.side_effect = values
num_rows = 3
columns = list('ABC')
expected_result = pd.DataFrame(values, columns=columns)
# Run
result = sampler._sample_model(model, num_rows, columns)
# Check
assert result.equals(expected_result)
qualified_mock.assert_called_once_with(model)
assert model.sample.call_args_list == [((3, ), ), ((3, ), ), ((3, ), )]
def test__sample_model(self, qualified_mock):
"""_sample_model sample the number of rows from the given model."""
# Setup
data_navigator = MagicMock(spec=DataNavigator)
modeler = MagicMock(spec=Modeler)
sampler = Sampler(data_navigator, modeler)
model = MagicMock()
values = np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])
qualified_mock.return_value = 'package.module.full_qualified_name'
model.sample.return_value = values
num_rows = 3
columns = list('ABC')
expected_result = pd.DataFrame(values, columns=columns)
# Run
result = sampler._sample_model(model, num_rows, columns)
# Check
assert result.equals(expected_result)
qualified_mock.assert_called_once_with(model)
model.sample.assert_called_once_with(3)
