Skip to content

Commit 51fd1fb

Browse files
committed
inserted an idea for testing in test_distributions_random.py
1 parent 42ffa0d commit 51fd1fb

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

pymc3/tests/test_distributions_random.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,13 +138,26 @@ def get_random_variable(self, shape, with_vector_params=False, name=None):
138138
else:
139139
return self.distribution(name, shape=shape, transform=None, **params)
140140

141+
def get_random_variable_default_params(self, shape, name=None):
142+
if name is None:
143+
name = self.distribution.__name__
144+
with self.model:
145+
if shape is None:
146+
return self.distribution(name, transform=None)
147+
else:
148+
return self.distribution(name, shape=shape, transform=None)
149+
141150
@staticmethod
142151
def sample_random_variable(random_variable, size):
143152
try:
144153
return random_variable.random(size=size)
145154
except AttributeError:
146155
return random_variable.distribution.random(size=size)
147156

157+
def test_sample_random_variable_default_params(self, size):
158+
rv = self.get_random_variable_default_params(None)
159+
self.sample_random_variable(rv, size)
160+
148161
@pytest.mark.parametrize('size', [None, 5, (4, 5)], ids=str)
149162
def test_scalar_parameter_shape(self, size):
150163
rv = self.get_random_variable(None)

0 commit comments

Comments
 (0)