Ejemplo n.º 1
0
    def test_rc_list(self):
        """Testing rc_list."""
        # Special case mp=1
        rc = rc_list(mp=1, sh=0.0, pbc=False, order="unit_cell")
        self.assert_equal(rc, [0.0])

        rc = rc_list(mp=1, sh=0.0, pbc=True, order="unit_cell")
        self.assert_equal(rc, [0.0, 1.0])

        rc = rc_list(mp=1, sh=0.0, pbc=False, order="bz")
        self.assert_equal(rc, [0.0])

        rc = rc_list(mp=1, sh=0.0, pbc=True, order="bz")
        self.assert_equal(rc, [0.0, 1.0])

        # Even mp
        rc = rc_list(mp=2, sh=0, pbc=False, order="unit_cell")
        self.assert_equal(rc, [0., 0.5])

        rc = rc_list(mp=2, sh=0, pbc=True, order="unit_cell")
        self.assert_equal(rc, [0., 0.5, 1.])

        rc = rc_list(mp=2, sh=0, pbc=False, order="bz")
        self.assert_equal(rc, [-0.5, 0.0])

        rc = rc_list(mp=2, sh=0, pbc=True, order="bz")
        self.assert_equal(rc, [-0.5, 0., 0.5])

        rc = rc_list(mp=2, sh=0.5, pbc=False, order="unit_cell")
        self.assert_equal(rc, [0.25, 0.75])

        rc = rc_list(mp=2, sh=0.5, pbc=True, order="unit_cell")
        self.assert_equal(rc, [0.25, 0.75, 1.25])

        rc = rc_list(mp=2, sh=0.5, pbc=False, order="bz")
        self.assert_equal(rc, [-0.25, 0.25])

        rc = rc_list(mp=2, sh=0.5, pbc=True, order="bz")
        self.assert_equal(rc, [-0.25, 0.25, 0.75])

        # Odd mp
        rc = rc_list(mp=3, sh=0, pbc=False, order="unit_cell")
        self.assert_almost_equal(rc, [0., 0.33333333, 0.66666667])

        rc = rc_list(mp=3, sh=0, pbc=True, order="unit_cell")
        self.assert_almost_equal(rc, [0., 0.33333333, 0.66666667, 1.])

        rc = rc_list(mp=3, sh=0, pbc=False, order="bz")
        self.assert_almost_equal(rc, [-0.33333333, 0., 0.33333333])

        rc = rc_list(mp=3, sh=0, pbc=True, order="bz")
        self.assert_almost_equal(rc, [-0.33333333, 0., 0.33333333, 0.66666667])

        rc = rc_list(mp=3, sh=0.5, pbc=False, order="unit_cell")
        self.assert_almost_equal(rc, [0.16666667, 0.5, 0.83333333])

        rc = rc_list(mp=3, sh=0.5, pbc=True, order="unit_cell")
        self.assert_almost_equal(rc, [0.16666667, 0.5, 0.83333333, 1.16666667])

        rc = rc_list(mp=3, sh=0.5, pbc=False, order="bz")
        self.assert_almost_equal(rc, [-0.5, -0.16666667, 0.16666667])

        rc = rc_list(mp=3, sh=0.5, pbc=True, order="bz")
        self.assert_almost_equal(rc, [-0.5, -0.16666667, 0.16666667, 0.5])
Ejemplo n.º 2
0
    def test_rc_list(self):
        """Testing rc_list."""
        # Special case mp=1
        rc = rc_list(mp=1, sh=0.0, pbc=False, order="unit_cell")
        self.assert_equal(rc, [0.0])

        rc = rc_list(mp=1, sh=0.0, pbc=True, order="unit_cell")
        self.assert_equal(rc, [0.0, 1.0])

        rc = rc_list(mp=1, sh=0.0, pbc=False, order="bz")
        self.assert_equal(rc, [0.0])

        rc = rc_list(mp=1, sh=0.0, pbc=True, order="bz")
        self.assert_equal(rc, [0.0, 1.0])

        # Even mp
        rc = rc_list(mp=2, sh=0, pbc=False, order="unit_cell")
        self.assert_equal(rc, [0.0, 0.5])

        rc = rc_list(mp=2, sh=0, pbc=True, order="unit_cell")
        self.assert_equal(rc, [0.0, 0.5, 1.0])

        rc = rc_list(mp=2, sh=0, pbc=False, order="bz")
        self.assert_equal(rc, [-0.5, 0.0])

        rc = rc_list(mp=2, sh=0, pbc=True, order="bz")
        self.assert_equal(rc, [-0.5, 0.0, 0.5])

        rc = rc_list(mp=2, sh=0.5, pbc=False, order="unit_cell")
        self.assert_equal(rc, [0.25, 0.75])

        rc = rc_list(mp=2, sh=0.5, pbc=True, order="unit_cell")
        self.assert_equal(rc, [0.25, 0.75, 1.25])

        rc = rc_list(mp=2, sh=0.5, pbc=False, order="bz")
        self.assert_equal(rc, [-0.25, 0.25])

        rc = rc_list(mp=2, sh=0.5, pbc=True, order="bz")
        self.assert_equal(rc, [-0.25, 0.25, 0.75])

        # Odd mp
        rc = rc_list(mp=3, sh=0, pbc=False, order="unit_cell")
        self.assert_almost_equal(rc, [0.0, 0.33333333, 0.66666667])

        rc = rc_list(mp=3, sh=0, pbc=True, order="unit_cell")
        self.assert_almost_equal(rc, [0.0, 0.33333333, 0.66666667, 1.0])

        rc = rc_list(mp=3, sh=0, pbc=False, order="bz")
        self.assert_almost_equal(rc, [-0.33333333, 0.0, 0.33333333])

        rc = rc_list(mp=3, sh=0, pbc=True, order="bz")
        self.assert_almost_equal(rc, [-0.33333333, 0.0, 0.33333333, 0.66666667])

        rc = rc_list(mp=3, sh=0.5, pbc=False, order="unit_cell")
        self.assert_almost_equal(rc, [0.16666667, 0.5, 0.83333333])

        rc = rc_list(mp=3, sh=0.5, pbc=True, order="unit_cell")
        self.assert_almost_equal(rc, [0.16666667, 0.5, 0.83333333, 1.16666667])

        rc = rc_list(mp=3, sh=0.5, pbc=False, order="bz")
        self.assert_almost_equal(rc, [-0.5, -0.16666667, 0.16666667])

        rc = rc_list(mp=3, sh=0.5, pbc=True, order="bz")
        self.assert_almost_equal(rc, [-0.5, -0.16666667, 0.16666667, 0.5])