black formatting

Author: michaelosthege

pymc3/distributions/distribution.py

@@ -943,9 +943,9 @@ def _draw_value(param, point=None, givens=None, size=None):
 
 
 def _is_one_d(dist_shape):
-    if hasattr(dist_shape, "dshape") and dist_shape.dshape in { (), }:
+    if hasattr(dist_shape, "dshape") and dist_shape.dshape in {()}:
         return True
-    elif hasattr(dist_shape, "shape") and dist_shape.shape in { (), }:
+    elif hasattr(dist_shape, "shape") and dist_shape.shape in {()}:
         return True
     return False
 

pymc3/tests/test_distributions_random.py

@@ -255,18 +255,24 @@ def test_scalar_distribution_shape(self, shape, size):
             exp_size = self.default_size if size is None else tuple(np.atleast_1d(size))
             expected = exp_size + exp_shape
             actual = np.shape(self.sample_random_variable(rv, size))
-            assert expected == actual, f"Sample size {size} from {shape}-shaped RV had shape {actual}. Expected: {expected}"
+            assert (
+                expected == actual
+            ), f"Sample size {size} from {shape}-shaped RV had shape {actual}. Expected: {expected}"
 
         @pytest.mark.parametrize("size", [None, ()], ids=str)
-        @pytest.mark.parametrize("shape", [None, (), (1,), (1, 1), (1, 2), (10, 11, 1), (9, 10, 2)], ids=str)
+        @pytest.mark.parametrize(
+            "shape", [None, (), (1,), (1, 1), (1, 2), (10, 11, 1), (9, 10, 2)], ids=str
+        )
         def test_scalar_sample_shape(self, shape, size):
             """ Draws samples of scalar [size] from a [shape] RV. """
             rv = self.get_random_variable(shape)
             exp_shape = self.default_shape if shape is None else tuple(np.atleast_1d(shape))
             exp_size = self.default_size if size is None else tuple(np.atleast_1d(size))
             expected = exp_size + exp_shape
             actual = np.shape(self.sample_random_variable(rv, size))
-            assert expected == actual, f"Sample size {size} from {shape}-shaped RV had shape {actual}. Expected: {expected}"
+            assert (
+                expected == actual
+            ), f"Sample size {size} from {shape}-shaped RV had shape {actual}. Expected: {expected}"
 
         @pytest.mark.parametrize("size", [None, 3, (4, 5)], ids=str)
         @pytest.mark.parametrize("shape", [None, 1, (10, 11, 1)], ids=str)
@@ -277,7 +283,9 @@ def test_vector_params(self, shape, size):
             exp_size = self.default_size if size is None else tuple(np.atleast_1d(size))
             expected = exp_size + exp_shape
             actual = np.shape(self.sample_random_variable(rv, size))
-            assert expected == actual, f"Sample size {size} from {shape}-shaped RV had shape {actual}. Expected: {expected}"
+            assert (
+                expected == actual
+            ), f"Sample size {size} from {shape}-shaped RV had shape {actual}. Expected: {expected}"
 
 
 class TestGaussianRandomWalk(BaseTestCases.BaseTestCase):