test column names

Author: ahartikainen

test/test_generate_quantities.py

@@ -85,27 +85,9 @@ def test_from_csv_files(self):
         )
 
         self.assertEqual(
-            list(bern_gqs.draws_pd(vars=column_names).columns),
+            list(bern_gqs.draws_pd(vars=['y_rep']).columns),
             column_names,
         )
-        self.assertEqual(
-            list(bern_gqs.draws_pd(vars=column_names[::-1]).columns),
-            column_names[::-1],
-        )
-        self.assertEqual(
-            list(bern_gqs.draws_pd(vars=column_names[:3]).columns),
-            column_names[:3],
-        )
-        self.assertEqual(
-            list(bern_gqs.draws_pd(vars=column_names[:3][::-1]).columns),
-            column_names[:3][::-1],
-        )
-        self.assertEqual(
-            list(bern_gqs.draws_pd(
-                vars=['y_rep[3]', 'y_rep[6]', 'y_rep[2]']
-            ).columns),
-            ['y_rep[3]', 'y_rep[6]', 'y_rep[2]'],
-        )
 
     def test_from_csv_files_bad(self):
         # gq model