Bugfixes

probabilistic-numerics · Aug 28, 2020 · 9414231 · 9414231
1 parent 4ce2dde
commit 9414231
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 4 deletions.
diff --git a/tests/test_diffeq/test_odesolver.py b/tests/test_diffeq/test_odesolver.py
@@ -3,7 +3,7 @@
 import numpy as np
 
 from probnum.diffeq import logistic, ODESolver, ConstantSteps
-from probnum.prob import RandomVariable, Dirac
+from probnum.random_variables import Dirac
 
 
 class MockODESolver(ODESolver):
@@ -17,7 +17,7 @@ def step(self, start, stop, current):
         x = current.mean()
         xnew = x + h * self.ivp(start, x)
         return (
-            RandomVariable(Dirac(xnew)),
+            Dirac(xnew),
             np.nan,
         )  # return nan as error estimate to ensure that it is not used
 
@@ -29,7 +29,7 @@ class ODESolverTestCase(unittest.TestCase):
     """
 
     def setUp(self):
-        y0 = RandomVariable(distribution=Dirac(0.3))
+        y0 = Dirac(0.3)
         ivp = logistic([0, 4], initrv=y0)
         self.solver = MockODESolver(ivp)
         self.step = 0.2

diff --git a/tests/test_core/__init__.py → tests/test_random_variables/__init__.py b/tests/test_core/__init__.py → tests/test_random_variables/__init__.py
diff --git a/..._core/test_random_variables/test_dirac.py → tests/test_random_variables/test_dirac.py b/..._core/test_random_variables/test_dirac.py → tests/test_random_variables/test_dirac.py
@@ -4,7 +4,6 @@
 import numpy as np
 
 from probnum import random_variables as rvs
-from probnum import utils as _utils
 
 
 class TestDirac(unittest.TestCase):

diff --git a/...core/test_random_variables/test_normal.py → tests/test_random_variables/test_normal.py b/...core/test_random_variables/test_normal.py → tests/test_random_variables/test_normal.py
diff --git a/..._random_variables/test_random_variable.py → ..._random_variables/test_random_variable.py b/..._random_variables/test_random_variable.py → ..._random_variables/test_random_variable.py