diff options
Diffstat (limited to 'tensorflow/contrib/distributions/python/kernel_tests/bijectors/softplus_test.py')
-rw-r--r-- | tensorflow/contrib/distributions/python/kernel_tests/bijectors/softplus_test.py | 25 |
1 files changed, 12 insertions, 13 deletions
diff --git a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/softplus_test.py b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/softplus_test.py index 2c58519fda..a7db1dcc6c 100644 --- a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/softplus_test.py +++ b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/softplus_test.py @@ -20,9 +20,8 @@ from __future__ import print_function import numpy as np -from tensorflow.contrib.distributions.python.ops.bijectors.bijector_test_util import assert_bijective_and_finite -from tensorflow.contrib.distributions.python.ops.bijectors.bijector_test_util import assert_scalar_congruency -from tensorflow.contrib.distributions.python.ops.bijectors.softplus import Softplus +from tensorflow.contrib.distributions.python.ops.bijectors import bijector_test_util +from tensorflow.contrib.distributions.python.ops.bijectors import softplus as softplus_lib from tensorflow.python.platform import test rng = np.random.RandomState(42) @@ -43,7 +42,7 @@ class SoftplusBijectorTest(test.TestCase): def testBijectorForwardInverseEventDimsZero(self): with self.test_session(): - bijector = Softplus(event_ndims=0) + bijector = softplus_lib.Softplus(event_ndims=0) self.assertEqual("softplus", bijector.name) x = 2 * rng.randn(2, 10) y = self._softplus(x) @@ -53,7 +52,7 @@ class SoftplusBijectorTest(test.TestCase): def testBijectorLogDetJacobianEventDimsZero(self): with self.test_session(): - bijector = Softplus(event_ndims=0) + bijector = softplus_lib.Softplus(event_ndims=0) y = 2 * rng.rand(2, 10) # No reduction needed if event_dims = 0. ildj = self._softplus_ildj_before_reduction(y) @@ -62,7 +61,7 @@ class SoftplusBijectorTest(test.TestCase): def testBijectorForwardInverseEventDimsOne(self): with self.test_session(): - bijector = Softplus(event_ndims=1) + bijector = softplus_lib.Softplus(event_ndims=1) self.assertEqual("softplus", bijector.name) x = 2 * rng.randn(2, 10) y = self._softplus(x) @@ -72,7 +71,7 @@ class SoftplusBijectorTest(test.TestCase): def testBijectorLogDetJacobianEventDimsOne(self): with self.test_session(): - bijector = Softplus(event_ndims=1) + bijector = softplus_lib.Softplus(event_ndims=1) y = 2 * rng.rand(2, 10) ildj_before = self._softplus_ildj_before_reduction(y) ildj = np.sum(ildj_before, axis=1) @@ -81,28 +80,28 @@ class SoftplusBijectorTest(test.TestCase): def testScalarCongruency(self): with self.test_session(): - bijector = Softplus(event_ndims=0) - assert_scalar_congruency( + bijector = softplus_lib.Softplus(event_ndims=0) + bijector_test_util.assert_scalar_congruency( bijector, lower_x=-2., upper_x=2.) def testBijectiveAndFinite32bit(self): with self.test_session(): - bijector = Softplus(event_ndims=0) + bijector = softplus_lib.Softplus(event_ndims=0) x = np.linspace(-20., 20., 100).astype(np.float32) y = np.logspace(-10, 10, 100).astype(np.float32) - assert_bijective_and_finite( + bijector_test_util.assert_bijective_and_finite( bijector, x, y, rtol=1e-2, atol=1e-2) def testBijectiveAndFinite16bit(self): with self.test_session(): - bijector = Softplus(event_ndims=0) + bijector = softplus_lib.Softplus(event_ndims=0) # softplus(-20) is zero, so we can't use such a large range as in 32bit. x = np.linspace(-10., 20., 100).astype(np.float16) # Note that float16 is only in the open set (0, inf) for a smaller # logspace range. The actual range was (-7, 4), so use something smaller # for the test. y = np.logspace(-6, 3, 100).astype(np.float16) - assert_bijective_and_finite( + bijector_test_util.assert_bijective_and_finite( bijector, x, y, rtol=1e-1, atol=1e-3) |