def test_normalize(self): for i in range(100): a = random.random() + i b = random.random() + i c = random.random() + i self.assertTrue( BaseUtils.equal(P3(a, b, c).normalize().length(), 1.0))
def test_tr_01(self): t = (Trajectory(StraightLine2(2.0, P2(1, 0), P2(0, 0))).add_arc_line( 0.95, False, 22.5).add_strait_line(1.5).add_arc_line( 0.95, False, 22.5).add_strait_line(2.0 + 2.2).add_arc_line( 0.95, True, 67.5).add_strait_line(1.5).add_arc_line( 0.95, True, 67.5).add_strait_line(2.2)) self.assertTrue(BaseUtils.equal(t.direct_at_end(), P2(0, -1)))
def test_change_length(self): for i in range(100): a = random.random() + i b = random.random() + i c = random.random() + i d = random.random() + i self.assertTrue( BaseUtils.equal(P3(a, b, c).change_length(d).length(), d))
def test_length(self): for i in range(100): a = random.random() + i b = random.random() + i c = random.random() + i self.assertTrue( BaseUtils.equal( P3(a, b, c).length(), float(np.sqrt(a**2 + b**2 + c**2))))
def test_mul2(self): for i in range(100): a = random.random() + i + 1 b = random.random() + i + 1 c = random.random() + i + 1 d = random.random() + i + 1 self.assertTrue( BaseUtils.equal(P2(a, b) * P2(c, d), a * c + b * d, msg="error"))
def test_mul2(self): for i in range(100): a = random.random() + i + 1 b = random.random() + i + 1 c = random.random() + i + 1 d = random.random() + i + 1 e = random.random() + i + 1 f = random.random() + i + 1 self.assertTrue( BaseUtils.equal(P3(a, b, e) * P3(c, d, f), a * c + b * d + e * f, msg="error"))
def test_angle_to(self): for i in range(100): a0 = random.random() * np.pi * 1.9 + 1e-6 b0 = random.random() * np.pi * 1.9 + 1e-6 a = max(a0, b0) b = min(a0, b0) p1 = P2(1.0) p2 = P2(1.0) p1 = p1.rotate(a) p2 = p2.rotate(b) diff = a - b # print(p1.angle_to(p2),diff) self.assertTrue( BaseUtils.equal( p2.angle_to(p1), diff #if diff >0 else 2 * np.pi - diff ))
def test_p2_copy(self): for i in range(100): p = P2(random.random() + i, random.random() + i) self.assertTrue(BaseUtils.equal(p, p.copy())) self.assertFalse(p is p.copy())
def test_p2_norm(self): p = P2(1, 1) self.assertTrue( BaseUtils.equal(p.normalize(), P2(np.sqrt(2) / 2, np.sqrt(2) / 2)))
def test_p2_len(self): p = P2(1, 1) self.assertTrue(BaseUtils.equal(p.length(), np.sqrt(2)))
def test_straight(self): s = StraightLine2(1, np.array([1.0, 1.0]), np.array([0.0, 0.0])) self.assertTrue(BaseUtils.equal(s.get_length(), 1))
def test_equal2(self): BaseUtils.equal(P2(), P2(1e-10), msg="不相等") with self.assertRaises(AssertionError): BaseUtils.equal(P2(), P2(1e-5), msg="不相等")
def test_equal(self): BaseUtils.equal(1, 1, msg="不相等") with self.assertRaises(AssertionError): BaseUtils.equal(1, 2, msg="不相等")
def test_angle_to_x_axis(self): for i in range(100): phi = random.random() * np.pi * 1.9 + 1e-6 p = P2(1.0) p = p.rotate(phi) self.assertTrue(BaseUtils.equal(p.angle_to_x_axis(), phi))
def test_p2_rotate(self): for phi in np.linspace(0.1, 2 * np.pi * 0.99, 100): p = P2(1.0) p = p.rotate(phi) self.assertTrue(BaseUtils.equal(p.angle_to_x_axis(), phi))
MM, ) cct = CCT( LocalCoordinateSystem.global_coordinate_system(), 0.95, 83 * MM + 15 * MM * 2, 67.5, [30.0, 80.0, 90.0, 90.0], 128, -9664, P2(0, 0), P2(128 * np.pi * 2, 67.5 / 180.0 * np.pi), ) # Plot2.plot_cct(cct) # Plot3.plot_cct(cct) # Plot2.show() # Plot3.show() m = cct.magnetic_field_at(P3()) print(m) import time s = time.time() for p in BaseUtils.linspace(P3(),P3(y=2),500): print(cct.magnetic_field_at(p)) print(time.time()-s)