def test_rho(self):
rs, rp = fresnel.rho(0, 1, 0, 1)
self.assertAlmostEqual(rs, 0, 7)
self.assertAlmostEqual(rp, 0, 7)
# testing brewster's angle
n1 = 1.0
n2 = 1.46
alpha_0 = np.arctan(n2/n1)
alpha_1 = fresnel.angle(alpha_0, n1, n2)
rs, rp = fresnel.rho(alpha_0, n1, alpha_1, n2)
self.assertAlmostEqual(rp, 0, 7)
def test_angle(self):
n = np.arange(1, 1.5, 10)
a = fresnel.angle(np.pi/4, 1.0, n)
self.assertEqual(len(n), len(a))
self.assertAlmostEqual(a[0], np.pi/4, 7)
