

def test_should_return_list_of_params(self):
    ret = sdg.generate_separated_model_parameters('continuous', 0.5, self.num_clusters, self.get_next_seed)
    assert isinstance(ret, list)
    assert (len(ret) == self.num_clusters)
    for entry in ret:
        assert isinstance(entry, dict)
        for key in entry.keys():
            assert (key in ['mu', 'rho'])
        assert (len(entry.keys()) == 2)
    ret = sdg.generate_separated_model_parameters('multinomial', 0.5, self.num_clusters, self.get_next_seed, distargs=self.distargs_multinomial)
    assert isinstance(ret, list)
    assert (len(ret) == self.num_clusters)
    for entry in ret:
        assert isinstance(entry, dict)
        for key in entry.keys():
            assert (key in ['weights'])
        assert (len(entry.keys()) == 1)

dataset/ETHPy150Open probcomp/crosscat/src/tests/unit_tests/test_synthetic_data_generator.py TestGenerateGeparatedGodelParameters.test_should_return_list_of_params/original