Fix lint

Author: ahartikainen

test/test_sample.py

@@ -740,12 +740,24 @@ def test_validate_good_run(self):
         self.assertEqual(fit.draws_pd(vars=['lp__', 'theta']).shape, (400, 2))
         self.assertEqual(fit.draws_pd(vars=['theta', 'lp__']).shape, (400, 2))
         self.assertEqual(fit.draws_pd(vars='theta').shape, (400, 1))
-        
-        self.assertEqual(fit.draws_pd(vars=['theta', 'lp__']).columns, ['theta', 'lp__'])
-        self.assertEqual(fit.draws_pd(vars=['lp__', 'theta']).columns, ['lp__', 'theta'])
-        self.assertEqual(fit.draws_pd(vars=fit.column_names).columns, fit.column_names)
-        self.assertEqual(fit.draws_pd(vars=fit.column_names[::-1]).columns, fit.column_names[::-1])
-        
+
+        self.assertEqual(
+            fit.draws_pd(vars=['theta', 'lp__']).columns,
+            ['theta', 'lp__']
+        )
+        self.assertEqual(
+            fit.draws_pd(vars=['lp__', 'theta']).columns,
+            ['lp__', 'theta']
+        )
+        self.assertEqual(
+            fit.draws_pd(vars=fit.column_names).columns,
+            fit.column_names
+        )
+        self.assertEqual(
+            fit.draws_pd(vars=fit.column_names[::-1]).columns,
+            fit.column_names[::-1]
+        )
+
         summary = fit.summary()
         self.assertIn('5%', list(summary.columns))
         self.assertIn('50%', list(summary.columns))