def test_simulate_observed_data_with_mock_mix_2(
mock_generate_start_factors_and_control_variables,
set_up_generate_datasets_mock_mix_2,
expected_dataset_mock_mix_2,
):
results = sd.simulate_datasets(**set_up_generate_datasets_mock_mix_2)
adfeq(results[0], expected_dataset_mock_mix_2["observed_data"], check_dtype=False)
def test_simulate_latent_data_with_mock(
mock_generate_start_factors_and_control_variables,
set_up_generate_datasets_mock,
expected_dataset_mock,
):
results = sd.simulate_datasets(**set_up_generate_datasets_mock)
adfeq(results[1], expected_dataset_mock["latent_data"], check_dtype=False)
def test_simulate_observed_data_2_mix(
set_up_generate_datasets_2_mix, expected_dataset_2_mix
):
obs_data = sd.simulate_datasets(**set_up_generate_datasets_2_mix)[0]
adfeq(obs_data, expected_dataset_2_mix["observed_data"], check_dtype=False)
def test_simulate_latent_data_2_mix(
set_up_generate_datasets_2_mix, expected_dataset_2_mix
):
latent_data = sd.simulate_datasets(**set_up_generate_datasets_2_mix)[1]
adfeq(latent_data, expected_dataset_2_mix["latent_data"], check_dtype=False)
def test_simulate_observed_data(set_up_generate_datasets, expected_dataset):
obs_data = sd.simulate_datasets(**set_up_generate_datasets)[0]
adfeq(obs_data, expected_dataset["observed_data"], check_dtype=False)
def test_simulate_latent_data(set_up_generate_datasets, expected_dataset):
latent_data = sd.simulate_datasets(**set_up_generate_datasets)[1]
adfeq(latent_data, expected_dataset["latent_data"], check_dtype=False)