Ejemplo n.º 1
0
    def test_read_write(self):
        op = OP('H2O.dat')
        ao_contrib = AO_Contrib(1, 'O', 'px', 100.0)
        orb = MOrbital()
        self.assertEqual(op.orb_list[4].contributions[0], ao_contrib)

        c1 = AO_Contrib(0, 'H', 's', 19.6)
        c2 = AO_Contrib(1, 'O', 'pz', 30.4)
        c3 = AO_Contrib(1, 'O', 'py', 30.4)
        c4 = AO_Contrib(2, 'H', 's', 19.6)
        contributions = [c1, c2, c3, c4]
        mo = MOrbital(2, -0.49905, 2, contributions)
        self.assertEqual(op[2], mo)

        op.write('tmp.csv', 'csv')
        op_dup = OP('tmp.csv')
        self.assertEqual(op, op_dup)
        os.remove('tmp.csv')
Ejemplo n.º 2
0
    def test_read_write(self):
        op = OP('H2O.dat')
        ao_contrib = AO_Contrib(1, 'O', 'px', 100.0)
        self.assertEqual(op.orb_list[4].contributions[0], ao_contrib)

        c1 = AO_Contrib(0, 'H', 's', 19.6)
        c2 = AO_Contrib(1, 'O', 'pz', 30.4)
        c3 = AO_Contrib(1, 'O', 'py', 30.4)
        c4 = AO_Contrib(2, 'H', 's', 19.6)
        contributions = [c1, c2, c3, c4]
        mo = MOrbital(2, None, -0.49905, 2, contributions)
        self.assertEqual(op[2], mo)

        op.write('tmp.csv', 'csv')
        op_dup = OP('tmp.csv')
        self.assertEqual(op, op_dup)
        os.remove('tmp.csv')

        # Test UKS open shell
        op = OP('H2O+.dat')
        ao_contrib = AO_Contrib(1, 'O', 'px', 100.0, 1)
        self.assertEqual(op.orb_list[3].contributions[0], ao_contrib)