metrolopy 0.6.3__py3-none-any.whl → 0.6.5__py3-none-any.whl

metrolopy/tests/test_misc.py

@@ -25,89 +25,103 @@ Check that known bugs are fixed.
 
 import metrolopy as uc
 import numpy as np
+import unittest
 
-def test_reduce():
-    a = uc.gummy(12.367,0.22,unit='mm/m')
-    a.reduce_unit()
-    assert abs(a.x - 0.012367) < 1e-15
-    assert abs(a.u - 0.00022) < 1e-15
-    assert a.unit is uc.one
-    
-def test_zero_derivative():
-    u = 0.00223
-    a = uc.gummy(0,u)
-    b = a**2
-    
-    assert b.u == 0
-    b.sim()
-    assert abs((np.sqrt(2)*u**2-b.usim)/b.usim) < 0.1
-    b.style = 'pmsim'
-    assert '?' not in b.tostring()
-    
-def test_triangular(plot=False):
-    a = uc.gummy(uc.TriangularDist(1,1))
-    a.sim()
-    assert abs((a.usim - 1/np.sqrt(6))/a.usim) < 0.1
-    if plot:
-        print('This should be a triangle with mode 1, lower limit 0 and upper limit 1:')
-        a.hist()
+
+class TestMisc(unittest.TestCase):
+
+    def test_reduce(self):
+        a = uc.gummy(12.367,0.22,unit='mm/m')
+        a.reduce_unit()
+        self.assertTrue(abs(a.x - 0.012367) < 1e-15)
+        self.assertTrue(abs(a.u - 0.00022) < 1e-15)
+        self.assertTrue(a.unit is uc.one)
         
-    a = uc.gummy(uc.TriangularDist(-2.5,left_width=0.5,right_width=1.5))
-    a.sim()
-    if plot:
-        print('this should be a triangle with mode -2.5, lower limit -3 and upper limit -1:')
-        a.hist()
+    def test_zero_derivative(self):
+        u = 0.00223
+        a = uc.gummy(0,u)
+        b = a**2
         
-    a = uc.gummy(uc.TriangularDist(1,lower_limit=0.5,upper_limit=1.5))
-    a.sim()
-    if plot:
-        print('this should be a triangle with mode 1, lower limit 0.5 and upper limit 1.5:')
-        a.hist()
+        self.assertTrue(b.u == 0)
+        b.sim()
+        self.assertTrue(abs((np.sqrt(2)*u**2-b.usim)/b.usim) < 0.1)
+        b.style = 'pmsim'
+        self.assertTrue('?' not in b.tostring())
         
-def test_zero_degc():
-    a = uc.gummy(0, 0.15,unit='degC')
-    a.style='xf'
-    assert a.tostring() == '0.00'
-    
-def test_ufrom_multiletter():
-    a = uc.gummy(10.0, 1, utype='A')
-    b = uc.gummy(20.0, 2, utype='DUT')
-    y = a - b
-    assert abs(y.ufrom('A') - 1) < 1e-15
-    assert abs(y.ufrom('DUT') - 2) < 1e-15
-    
-def test_uniform_params():
-    # issue 33
-    g = uc.gummy(uc.UniformDist(lower_limit=6,upper_limit=9),unit='pA')
-    g.sim()
-    assert abs(min(g.simdata) - 6) < 0.01
-    assert abs(max(g.simdata) - 9) < 0.01
+    def test_triangular(self,plot=False):
+        a = uc.gummy(uc.TriangularDist(1,1))
+        a.sim()
+        self.assertTrue(abs((a.usim - 1/np.sqrt(6))/a.usim) < 0.1)
+        if plot:
+            print('This should be a triangle with mode 1, lower limit 0 and upper limit 1:')
+            a.hist()
+            
+        a = uc.gummy(uc.TriangularDist(-2.5,left_width=0.5,right_width=1.5))
+        a.sim()
+        if plot:
+            print('this should be a triangle with mode -2.5, lower limit -3 and upper limit -1:')
+            a.hist()
+            
+        a = uc.gummy(uc.TriangularDist(1,lower_limit=0.5,upper_limit=1.5))
+        a.sim()
+        if plot:
+            print('this should be a triangle with mode 1, lower limit 0.5 and upper limit 1.5:')
+            a.hist()
+            
+    def test_zero_degc(self):
+        a = uc.gummy(0, 0.15,unit='degC')
+        a.style='xf'
+        self.assertTrue(a.tostring() == '0.00')
+        
+    def test_ufrom_multiletter(self):
+        a = uc.gummy(10.0, 1, utype='A')
+        b = uc.gummy(20.0, 2, utype='DUT')
+        y = a - b
+        self.assertTrue(abs(y.ufrom('A') - 1) < 1e-15)
+        self.assertTrue(abs(y.ufrom('DUT') - 2) < 1e-15)
+        
+    def test_uniform_params(self):
+        # issue 33
+        g = uc.gummy(uc.UniformDist(lower_limit=6,upper_limit=9),unit='pA')
+        g.sim()
+        self.assertTrue(abs(min(g.simdata) - 6) < 0.01)
+        self.assertTrue(abs(max(g.simdata) - 9) < 0.01)
+        
+        g = uc.gummy(uc.UniformDist(center=7.5,half_width=1.5),unit='pA')
+        g.sim()
+        self.assertTrue(abs(min(g.simdata) - 6) < 0.01)
+        self.assertTrue(abs(max(g.simdata) - 9) < 0.01)
+        
+    def test_mean_b(self):
+        # issue 32
+        import numpy as np
+        from scipy.stats import t
+        from scipy import special
     
-    g = uc.gummy(uc.UniformDist(center=7.5,half_width=1.5),unit='pA')
-    g.sim()
-    assert abs(min(g.simdata) - 6) < 0.01
-    assert abs(max(g.simdata) - 9) < 0.01
+        con = special.erf(1/np.sqrt(2))
+        def u(x):
+            return t.ppf(1-(1-con)/2, len(x)-1)*x.std(ddof=1)/np.sqrt(len(x))
     
-def test_mean_b():
-    # issue 32
-    import numpy as np
-    from scipy.stats import t
-    from scipy import special
-
-    con = special.erf(1/np.sqrt(2))
-    def u(x):
-        return t.ppf(1-(1-con)/2, len(x)-1)*x.std(ddof=1)/np.sqrt(len(x))
-
-    x = np.random.rand(5)
-    g = uc.mean(x)
-    g.p = 'ssd'
-    assert abs(con-g.p) < 1e-6
-    assert abs(g.U - u(x)) < 1e-6
-    try:
-        uc.gummy.bayesian = True
-        gg = uc.mean(x)
-        assert abs(gg.u - np.sqrt(2)*g.u) < 1e-6
-        gg.p = 'ssd'
-        assert abs(gg.U - u(x)) < 1e-6
-    finally:
-        uc.gummy.bayesian = False
+        x = np.random.rand(5)
+        g = uc.mean(x)
+        g.p = 'ssd'
+        self.assertTrue(abs(con-g.p) < 1e-6)
+        self.assertTrue(abs(g.U - u(x)) < 1e-6)
+        try:
+            uc.gummy.bayesian = True
+            gg = uc.mean(x)
+            self.assertTrue(abs(gg.u - np.sqrt(2)*g.u) < 1e-6)
+            gg.p = 'ssd'
+            self.assertTrue(abs(gg.U - u(x)) < 1e-6)
+        finally:
+            uc.gummy.bayesian = False
+            
+    def test_uniform_hw(self):
+        # issue 37
+        a = uc.gummy(uc.UniformDist(upper_limit=1,lower_limit=-1))
+        b = uc.gummy(uc.UniformDist(center=0,half_width=1))
+        self.assertTrue(abs(a.u - b.u) < 1e-15)
+        
+        
+if __name__ == '__main__':
+    unittest.main()