moved transforms into their own test file

Author: phobson

probscale/tests/test_probscale/test_probscale.py

@@ -73,8 +73,7 @@ def test_cdf(self):
         )
 
 
-class Mixin_ProbFormatter_sig_figs(object):
-    fmt = probscale.PctFormatter()
+class Mixin_Check_Formatter_sig_figs(object):
     def teardown(self):
         pass
 
@@ -109,6 +108,9 @@ def test_forceint(self):
             self.known_int
         )
 
+
+class Mixin_Check_PctFormatter_sig_figs(Mixin_Check_Formatter_sig_figs):
+    fmt = probscale.PctFormatter()
     def test__call__(self):
         nt.assert_equal(self.fmt(0.0301), '0.03')
         nt.assert_equal(self.fmt(0.2), '0.2')
@@ -121,7 +123,20 @@ def test__call__(self):
         nt.assert_equal(self.fmt(99.99), '99.99')
 
 
-class Test_ProbFormatter_sig_figs_gt1(Mixin_ProbFormatter_sig_figs):
+class Mixin_Check_ProbFormatter_sig_figs(Mixin_Check_Formatter_sig_figs):
+    fmt = probscale.ProbFormatter()
+    def test__call__(self):
+        nt.assert_equal(self.fmt(0.000301), '0.0003')
+        nt.assert_equal(self.fmt(0.001), '0.001')
+        nt.assert_equal(self.fmt(0.10), '0.10')
+        nt.assert_equal(self.fmt(0.05), '0.05')
+        nt.assert_equal(self.fmt(0.50), '0.50')
+        nt.assert_equal(self.fmt(0.99), '0.99')
+        nt.assert_equal(self.fmt(0.991), '0.991')
+        nt.assert_equal(self.fmt(0.9999), '0.9999')
+
+
+class Test_PctFormatter_sig_figs_gt1(Mixin_Check_PctFormatter_sig_figs):
     def setup(self):
         self.x = 1234.56
         self.known_3 = '1,230'
@@ -132,7 +147,7 @@ def setup(self):
         self.factor = 10**5
 
 
-class Test_ProbFormatter_sig_figs_lt1(Mixin_ProbFormatter_sig_figs):
+class Test_PctFormatter_sig_figs_lt1(Mixin_Check_PctFormatter_sig_figs):
     def setup(self):
         self.x = 0.123456
         self.known_3 = '0.123'
@@ -213,3 +228,12 @@ def test_the_scale_beta():
     fig.tight_layout()
 
 
+class Test_ProbFormatter_sig_figs(Mixin_Check_ProbFormatter_sig_figs):
+    def setup(self):
+        self.x = 0.123456
+        self.known_3 = '0.123'
+        self.known_4 = '0.1235'
+        self.known_8 = '0.12345600'
+        self.known_exp3 = '1.23e-07'
+        self.known_int = '0'
+        self.factor = 10**-6

probscale/tests/test_probscale/test_transforms.py

@@ -0,0 +1,52 @@
+import matplotlib
+matplotlib.use('agg')
+
+import nose.tools as nt
+
+from probscale import probscale
+
+
+class Mixin_Transform(object):
+    known_input_dims = 1
+    known_output_dims = 1
+    known_is_separable = True
+    known_has_inverse = True
+
+    def test_input_dims(self):
+        nt.assert_true(hasattr(self.trans, 'input_dims'))
+        nt.assert_equal(self.trans.input_dims, self.known_input_dims)
+
+    def test_output_dims(self):
+        nt.assert_true(hasattr(self.trans, 'output_dims'))
+        nt.assert_equal(self.trans.output_dims, self.known_output_dims)
+
+    def test_is_separable(self):
+        nt.assert_true(hasattr(self.trans, 'is_separable'))
+        nt.assert_equal(self.trans.is_separable, self.known_is_separable)
+
+    def test_has_inverse(self):
+        nt.assert_true(hasattr(self.trans, 'has_inverse'))
+        nt.assert_equal(self.trans.has_inverse, self.known_has_inverse)
+
+    def test_dist(self):
+        nt.assert_true(hasattr(self.trans, 'dist'))
+        nt.assert_equal(self.trans.dist, probscale._minimal_norm)
+
+    def test_transform_non_affine(self):
+        nt.assert_true(hasattr(self.trans, 'transform_non_affine'))
+        nt.assert_almost_equal(self.trans.transform_non_affine(0.5), self.known_tras_na)
+
+    def test_inverted(self):
+        nt.assert_true(hasattr(self.trans, 'inverted'))
+
+
+class Test_ProbTransform(Mixin_Transform):
+    def setup(self):
+        self.trans = probscale.ProbTransform(probscale._minimal_norm)
+        self.known_tras_na = -2.569150498
+
+
+class Test_InvertedProbTransform(Mixin_Transform):
+    def setup(self):
+        self.trans = probscale.InvertedProbTransform(probscale._minimal_norm)
+        self.known_tras_na = 69.1464492