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)