

def test_should_return_array_of_proper_size(self):
    columns_list = [0]
    X = sdg.predictive_columns(self.M_c, self.X_L, self.X_D, columns_list)
    assert isinstance(X, numpy.ndarray)
    assert (X.shape[0] == self.num_rows)
    assert (X.shape[1] == len(columns_list))
    columns_list = [0, 1]
    X = sdg.predictive_columns(self.M_c, self.X_L, self.X_D, columns_list)
    assert isinstance(X, numpy.ndarray)
    assert (X.shape[0] == columns_list.num_rows)
    assert (X.shape[1] == len(columns_list))

dataset/ETHPy150Open probcomp/crosscat/src/tests/unit_tests/test_synthetic_data_generator.py TestPredictiveColumns.test_should_return_array_of_proper_size/VarMisuse@124/160 `self`->`columns_list`