From f0d976d32ce746003e89f8da353af0f66536a372 Mon Sep 17 00:00:00 2001 From: Philipp Arras Date: Wed, 31 Jul 2019 16:30:50 +0200 Subject: [PATCH] Streamlining --- test/test_linearization.py | 17 +---------------- 1 file changed, 1 insertion(+), 16 deletions(-) diff --git a/test/test_linearization.py b/test/test_linearization.py index e386638d..a1b083e3 100644 --- a/test/test_linearization.py +++ b/test/test_linearization.py @@ -32,29 +32,14 @@ def jt(lin, check): assert_allclose(_lin2grad(lin), check) -def test_lin(): +def test_special_gradients(): dom = ift.UnstructuredDomain((1,)) f = ift.full(dom, 2.4) var = ift.Linearization.make_var(f) s = f.local_data - jt(var.exp(), np.exp(s)) jt(var.clip(0, 10), np.ones_like(s)) jt(var.clip(-1, 0), np.zeros_like(s)) - jt(var.sqrt(), .5/np.sqrt(s)) - jt(var.sin(), np.cos(s)) - jt(var.cos(), -np.sin(s)) - jt(var.tan(), 1/np.cos(s)**2) - jt(var.sinc(), (np.cos(np.pi*s) - np.sinc(s))/s) - jt(var.log(), 1/s) - jt(var.sinh(), np.cosh(s)) - jt(var.cosh(), np.sinh(s)) - jt(var.tanh(), 1 - np.tanh(s)**2) - jt(var.absolute(), np.ones_like(s)) - jt(var.one_over(), -1/s**2) - - sigmoid = 0.5*(1 + var.tanh()) - jt(var.sigmoid(), sigmoid.jac(ift.full(sigmoid.domain, 1.)).local_data) assert_allclose( _lin2grad(ift.Linearization.make_var(0*f).sinc()), np.zeros(s.shape)) -- GitLab