clean up test_distribution
@@ -169,6 +169,7 @@ class DiscreteDistributionTests(unittest.TestCase):
self.assertAlmostEqual(ce9[3], 9.518015322143837)
+
class MatrixDiscreteDistributionTests(unittest.TestCase):
"""
Tests matrix-valued discrete distribution.
@@ -177,7 +178,10 @@ class MatrixDiscreteDistributionTests(unittest.TestCase):
def setUp(self):
self.draw_1 = np.array(
- [[[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]], [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],]
+ [
+ [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
+ [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],
+ ]
)
self.draw_2 = -1 * self.draw_1
