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] == self.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/original