test.py 文件源码-python代码片段

test.py 文件源码

python

阅读 32 收藏 0 点赞 0 评论 0

项目：electrostatics 作者: tomduck 项目源码文件源码

def test_E_2(self):
        """Tests the electric field 2."""
        E = self.charge2.E
        self.assertTrue(isclose(E([1, 0]), [3*sqrt(2), 0]).all())
        self.assertTrue(isclose(E([-1, 0]), [-3*sqrt(2), 0]).all())
        self.assertTrue(isclose(E([2, 1]),
                                array([1/sqrt(2), 1-1/sqrt(2)])*1.5).all())
        self.assertTrue(isclose(E([-2, 1]),
                                array([-1/sqrt(2), 1-1/sqrt(2)])*1.5).all())
        self.assertTrue(isclose(E([2, -1]),
                                array([1/sqrt(2), -1+1/sqrt(2)])*1.5).all())
        self.assertTrue(isclose(E([-2, -1]),
                                array([-1/sqrt(2), -1+1/sqrt(2)])*1.5).all())

        self.assertTrue(isclose(E([0, 2]), [0, 2]).all())
        self.assertTrue(isclose(E([0, -2]), [0, -2]).all())