added a few more tests

andim · May 23, 2017 · 2ee8f9c · 2ee8f9c
1 parent d26ce39
commit 2ee8f9c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/noisyopt/tests/test_noisyopt.py b/noisyopt/tests/test_noisyopt.py
@@ -19,6 +19,12 @@ def callback(x):
     res = noisyopt.minimizeCompass(quadratic, np.asarray([0.5, 1.0]), deltatol=deltatol,
                             errorcontrol=False, callback=callback)
 
+    # test with scaling
+    res = noisyopt.minimizeCompass(quadratic, np.asarray([0.5, 1.0]), deltatol=deltatol,
+                            scaling=np.array([0.1, 1.0]),
+                            errorcontrol=False)
+    npt.assert_allclose(res.x, [0.0, 0.0], atol=deltatol)
+
     # test with output
     res = noisyopt.minimizeCompass(quadratic, np.asarray([0.5, 1.0]), deltatol=deltatol,
                             errorcontrol=False, disp=True)
@@ -224,6 +230,11 @@ def test_DifferenceFunction():
     difffunc = noisyopt.DifferenceFunction(lambda x: x+1, lambda x: x+2)
     d, dse = difffunc(0)
     assert d == -1.0
+    d, dse = difffunc(1)
+    assert d == -1.0
+    # again with same value to test caching
+    d, dse = difffunc(0)
+    assert d == -1.0
 
 def test_memoized():
     # memoize simple function