Skip to content

Commit ef975dc

Browse files
authored
Check draws_pd order
1 parent a924d78 commit ef975dc

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

test/test_sample.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -742,21 +742,13 @@ def test_validate_good_run(self):
742742
self.assertEqual(fit.draws_pd(vars='theta').shape, (400, 1))
743743

744744
self.assertEqual(
745-
fit.draws_pd(vars=['theta', 'lp__']).columns,
745+
list(fit.draws_pd(vars=['theta', 'lp__']).columns),
746746
['theta', 'lp__']
747747
)
748748
self.assertEqual(
749-
fit.draws_pd(vars=['lp__', 'theta']).columns,
749+
list(fit.draws_pd(vars=['lp__', 'theta']).columns),
750750
['lp__', 'theta']
751751
)
752-
self.assertEqual(
753-
fit.draws_pd(vars=fit.column_names).columns,
754-
fit.column_names
755-
)
756-
self.assertEqual(
757-
fit.draws_pd(vars=fit.column_names[::-1]).columns,
758-
fit.column_names[::-1]
759-
)
760752

761753
summary = fit.summary()
762754
self.assertIn('5%', list(summary.columns))

0 commit comments

Comments
 (0)