cell bug fix 1

Author: felix5572

tests/test_cell_to_low_triangle.py

@@ -1,18 +1,18 @@
 import os
 import numpy as np
 import unittest
-from dpdata.cp2k.output import cell_to_low_triangle
+from context import dpdata
 
-class CellToLowTriangle(unittest.TestCase):
+class TestCellToLowTriangle(unittest.TestCase):
     def test_func1(self):
-        cell_1 = cell_to_low_triangle(6,6,6,np.pi*1/2, np.pi*1/2, np.pi*1/2)
+        cell_1 = dpdata.cp2k.output.cell_to_low_triangle(6,6,6,np.pi*1/2, np.pi*1/2, np.pi*1/2)
         cell_2 = np.asarray([[6,0,0],[0,6,0],[0,0,6]])
         for ii in range(3):
             for jj in range(3):
                 self.assertAlmostEqual(cell_1[ii,jj], cell_2[ii,jj], places=6)
 
     def test_func2(self):
-        cell_1 = cell_to_low_triangle(6,6,6,np.pi*1/3, np.pi*1/3, np.pi*1/3)
+        cell_1 = dpdata.cp2k.output.cell_to_low_triangle(6,6,6,np.pi*1/3, np.pi*1/3, np.pi*1/3)
         cell_2 = np.asarray([
             [6,0,0],
             [3,3*np.sqrt(3),0],
@@ -23,12 +23,12 @@ def test_func2(self):
 
     def test_func3(self):
         with self.assertRaises(Exception) as c:
-            cell_to_low_triangle(0.1,6,6,np.pi*1/2,np.pi*1/2,np.pi*1/2)
+            dpdata.cp2k.output.cell_to_low_triangle(0.1,6,6,np.pi*1/2,np.pi*1/2,np.pi*1/2)
         self.assertTrue("A==0.1" in str(c.exception))
 
     def test_func4(self):
         with self.assertRaises(Exception) as c:
-            cell_to_low_triangle(6,6,6,np.pi*3/180,np.pi*1/2,np.pi*1/2)
+            dpdata.cp2k.output.cell_to_low_triangle(6,6,6,np.pi*3/180,np.pi*1/2,np.pi*1/2)
         self.assertTrue("alpha" in str(c.exception))
 
 if __name__ == '__main__':