diff --git a/source/tests/test_data_modifier.py b/source/tests/test_data_modifier.py index c824262eb1..829a589d7e 100644 --- a/source/tests/test_data_modifier.py +++ b/source/tests/test_data_modifier.py @@ -28,7 +28,7 @@ INPUT = os.path.join(modifier_datapath, 'dipole.json') -class TestDataModifier (unittest.TestCase) : +class TestDataModifier (tf.test.TestCase) : def setUp(self): # with tf.variable_scope('load', reuse = False) : @@ -74,7 +74,7 @@ def _setUp(self): model.build (data) # freeze the graph - with tf.Session() as sess: + with self.test_session() as sess: init_op = tf.global_variables_initializer() sess.run(init_op) graph = tf.get_default_graph() diff --git a/source/tests/test_data_modifier_shuffle.py b/source/tests/test_data_modifier_shuffle.py index 397cb7fa5b..bd4ab58132 100644 --- a/source/tests/test_data_modifier_shuffle.py +++ b/source/tests/test_data_modifier_shuffle.py @@ -29,7 +29,7 @@ modifier_datapath = 'data_modifier' -class TestDataModifier (unittest.TestCase) : +class TestDataModifier (tf.test.TestCase) : def setUp(self): # with tf.variable_scope('load', reuse = False) : @@ -78,7 +78,7 @@ def _setUp(self): model.build (data) # freeze the graph - with tf.Session() as sess: + with self.test_session() as sess: init_op = tf.global_variables_initializer() sess.run(init_op) graph = tf.get_default_graph() diff --git a/source/tests/test_descrpt_nonsmth.py b/source/tests/test_descrpt_nonsmth.py index 1b99934d12..decb18e053 100644 --- a/source/tests/test_descrpt_nonsmth.py +++ b/source/tests/test_descrpt_nonsmth.py @@ -24,8 +24,9 @@ class Inter(): def setUp (self, data, comp = 0, - pbc = True) : - self.sess = tf.Session() + pbc = True, + sess = None) : + self.sess = sess self.data = data self.natoms = self.data.get_natoms() self.ntypes = self.data.get_ntypes() @@ -155,18 +156,18 @@ def comp_v_dw (self, -class TestNonSmooth(Inter, unittest.TestCase): +class TestNonSmooth(Inter, tf.test.TestCase): # def __init__ (self, *args, **kwargs): # self.places = 5 # data = Data() # Inter.__init__(self, data) - # unittest.TestCase.__init__(self, *args, **kwargs) + # tf.test.TestCase.__init__(self, *args, **kwargs) # self.controller = object() def setUp(self): self.places = 5 data = Data() - Inter.setUp(self, data) + Inter.setUp(self, data, sess=self.test_session().__enter__()) def test_force (self) : force_test(self, self, suffix = '_se') @@ -181,13 +182,13 @@ def test_virial_dw (self) : virial_dw_test(self, self, suffix = '_se') -class TestLFPbc(unittest.TestCase): +class TestLFPbc(tf.test.TestCase): def test_pbc(self): data = Data() inter0 = Inter() inter1 = Inter() - inter0.setUp(data, pbc = True) - inter1.setUp(data, pbc = False) + inter0.setUp(data, pbc = True, sess=self.test_session().__enter__()) + inter1.setUp(data, pbc = False, sess=self.test_session().__enter__()) inter0.net_w_i = np.copy(np.ones(inter0.ndescrpt)) inter1.net_w_i = np.copy(np.ones(inter1.ndescrpt)) @@ -227,8 +228,8 @@ def test_pbc_small_box(self): data1 = Data(box_scale = 2) inter0 = Inter() inter1 = Inter() - inter0.setUp(data0, pbc = True) - inter1.setUp(data1, pbc = False) + inter0.setUp(data0, pbc = True, sess=self.test_session().__enter__()) + inter1.setUp(data1, pbc = False, sess=self.test_session().__enter__()) inter0.net_w_i = np.copy(np.ones(inter0.ndescrpt)) inter1.net_w_i = np.copy(np.ones(inter1.ndescrpt)) diff --git a/source/tests/test_descrpt_se_a_type.py b/source/tests/test_descrpt_se_a_type.py index c35d952157..c6f3cb5a19 100644 --- a/source/tests/test_descrpt_se_a_type.py +++ b/source/tests/test_descrpt_se_a_type.py @@ -16,7 +16,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -110,7 +110,7 @@ def test_descriptor_two_sides(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [model_dout] = sess.run([dout], feed_dict = feed_dict_test) @@ -219,7 +219,7 @@ def test_descriptor_one_side(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [model_dout] = sess.run([dout], feed_dict = feed_dict_test) diff --git a/source/tests/test_descrpt_se_ar.py b/source/tests/test_descrpt_se_ar.py index 03026cff97..9ebb085bab 100644 --- a/source/tests/test_descrpt_se_ar.py +++ b/source/tests/test_descrpt_se_ar.py @@ -22,8 +22,9 @@ class Inter(): def setUp (self, - data) : - self.sess = tf.Session() + data, + sess = None) : + self.sess = sess self.data = data self.natoms = self.data.get_natoms() self.ntypes = self.data.get_ntypes() @@ -94,17 +95,17 @@ def comp_ef (self, return energy, force, virial -class TestDescrptAR(Inter, unittest.TestCase): +class TestDescrptAR(Inter, tf.test.TestCase): # def __init__ (self, *args, **kwargs): # data = Data() # Inter.__init__(self, data) - # unittest.TestCase.__init__(self, *args, **kwargs) + # tf.test.TestCase.__init__(self, *args, **kwargs) # self.controller = object() def setUp(self): self.places = 5 data = Data() - Inter.setUp(self, data) + Inter.setUp(self, data, sess=self.test_session().__enter__()) def test_force (self) : force_test(self, self, suffix = '_se_ar') diff --git a/source/tests/test_descrpt_se_r.py b/source/tests/test_descrpt_se_r.py index 4d6222a728..bf002736b0 100644 --- a/source/tests/test_descrpt_se_r.py +++ b/source/tests/test_descrpt_se_r.py @@ -23,8 +23,9 @@ class Inter(): def setUp (self, data, - pbc = True) : - self.sess = tf.Session() + pbc = True, + sess = None) : + self.sess = sess self.data = data self.natoms = self.data.get_natoms() self.ntypes = self.data.get_ntypes() @@ -136,17 +137,17 @@ def comp_v_dw (self, -class TestSmooth(Inter, unittest.TestCase): +class TestSmooth(Inter, tf.test.TestCase): # def __init__ (self, *args, **kwargs): # data = Data() # Inter.__init__(self, data) - # unittest.TestCase.__init__(self, *args, **kwargs) + # tf.test.TestCase.__init__(self, *args, **kwargs) # self.controller = object() def setUp(self): self.places = 5 data = Data() - Inter.setUp(self, data) + Inter.setUp(self, data, sess=self.test_session().__enter__()) def test_force (self) : force_test(self, self, suffix = '_se_r') @@ -161,13 +162,13 @@ def test_virial_dw (self) : virial_dw_test(self, self, suffix = '_se_r') -class TestSeRPbc(unittest.TestCase): +class TestSeRPbc(tf.test.TestCase): def test_pbc(self): data = Data() inter0 = Inter() inter1 = Inter() - inter0.setUp(data, pbc = True) - inter1.setUp(data, pbc = False) + inter0.setUp(data, pbc = True, sess=self.test_session().__enter__()) + inter1.setUp(data, pbc = False, sess=self.test_session().__enter__()) inter0.net_w_i = np.copy(np.ones(inter0.ndescrpt)) inter1.net_w_i = np.copy(np.ones(inter1.ndescrpt)) @@ -208,8 +209,8 @@ def test_pbc_small_box(self): data1 = Data(box_scale = 2) inter0 = Inter() inter1 = Inter() - inter0.setUp(data0, pbc = True) - inter1.setUp(data1, pbc = False) + inter0.setUp(data0, pbc = True, sess=self.test_session().__enter__()) + inter1.setUp(data1, pbc = False, sess=self.test_session().__enter__()) inter0.net_w_i = np.copy(np.ones(inter0.ndescrpt)) inter1.net_w_i = np.copy(np.ones(inter1.ndescrpt)) diff --git a/source/tests/test_descrpt_sea_ef.py b/source/tests/test_descrpt_sea_ef.py index 43b95f1cbc..b15aaaa7b0 100644 --- a/source/tests/test_descrpt_sea_ef.py +++ b/source/tests/test_descrpt_sea_ef.py @@ -23,8 +23,9 @@ class Inter(): def setUp (self, data, - pbc = True) : - self.sess = tf.Session() + pbc = True, + sess = None) : + self.sess = sess self.data = data self.natoms = self.data.get_natoms() self.ntypes = self.data.get_ntypes() @@ -148,17 +149,17 @@ def comp_v_dw (self, -class TestSmooth(Inter, unittest.TestCase): +class TestSmooth(Inter, tf.test.TestCase): # def __init__ (self, *args, **kwargs): # data = Data() # Inter.__init__(self, data) - # unittest.TestCase.__init__(self, *args, **kwargs) + # tf.test.TestCase.__init__(self, *args, **kwargs) # self.controller = object() def setUp(self): self.places = 5 data = Data() - Inter.setUp(self, data) + Inter.setUp(self, data, sess=self.test_session().__enter__()) def test_force (self) : force_test(self, self, suffix = '_sea_ef') diff --git a/source/tests/test_descrpt_sea_ef_para.py b/source/tests/test_descrpt_sea_ef_para.py index 0e94147161..213fc17930 100644 --- a/source/tests/test_descrpt_sea_ef_para.py +++ b/source/tests/test_descrpt_sea_ef_para.py @@ -23,8 +23,9 @@ class Inter(): def setUp (self, data, - pbc = True) : - self.sess = tf.Session() + pbc = True, + sess = None) : + self.sess = sess self.data = data self.natoms = self.data.get_natoms() self.ntypes = self.data.get_ntypes() @@ -148,17 +149,17 @@ def comp_v_dw (self, -class TestSmooth(Inter, unittest.TestCase): +class TestSmooth(Inter, tf.test.TestCase): # def __init__ (self, *args, **kwargs): # data = Data() # Inter.__init__(self, data) - # unittest.TestCase.__init__(self, *args, **kwargs) + # tf.test.TestCase.__init__(self, *args, **kwargs) # self.controller = object() def setUp(self): self.places = 5 data = Data() - Inter.setUp(self, data) + Inter.setUp(self, data, sess=self.test_session().__enter__()) def test_force (self) : force_test(self, self, suffix = '_sea_ef_para') diff --git a/source/tests/test_descrpt_sea_ef_rot.py b/source/tests/test_descrpt_sea_ef_rot.py index 34c7434a66..f172ec2771 100644 --- a/source/tests/test_descrpt_sea_ef_rot.py +++ b/source/tests/test_descrpt_sea_ef_rot.py @@ -12,9 +12,9 @@ from deepmd.descriptor import DescrptSeA from deepmd.descriptor import DescrptSeAEfLower -class TestEfRot(unittest.TestCase): +class TestEfRot(tf.test.TestCase): def setUp(self): - self.sess = tf.Session() + self.sess = self.test_session().__enter__() self.natoms = [5, 5, 2, 3] self.ntypes = 2 self.sel_a = [12,24] diff --git a/source/tests/test_descrpt_sea_ef_vert.py b/source/tests/test_descrpt_sea_ef_vert.py index a0d160e966..8ba82709e3 100644 --- a/source/tests/test_descrpt_sea_ef_vert.py +++ b/source/tests/test_descrpt_sea_ef_vert.py @@ -23,8 +23,9 @@ class Inter(): def setUp (self, data, - pbc = True) : - self.sess = tf.Session() + pbc = True, + sess = None) : + self.sess = sess self.data = data self.natoms = self.data.get_natoms() self.ntypes = self.data.get_ntypes() @@ -148,17 +149,17 @@ def comp_v_dw (self, -class TestSmooth(Inter, unittest.TestCase): +class TestSmooth(Inter, tf.test.TestCase): # def __init__ (self, *args, **kwargs): # data = Data() # Inter.__init__(self, data) - # unittest.TestCase.__init__(self, *args, **kwargs) + # tf.test.TestCase.__init__(self, *args, **kwargs) # self.controller = object() def setUp(self): self.places = 5 data = Data() - Inter.setUp(self, data) + Inter.setUp(self, data, sess=self.test_session().__enter__()) def test_force (self) : force_test(self, self, suffix = '_sea_ef_vert') diff --git a/source/tests/test_descrpt_smooth.py b/source/tests/test_descrpt_smooth.py index 05cb14b26e..adb5beb354 100644 --- a/source/tests/test_descrpt_smooth.py +++ b/source/tests/test_descrpt_smooth.py @@ -23,8 +23,9 @@ class Inter(): def setUp (self, data, - pbc = True) : - self.sess = tf.Session() + pbc = True, + sess = None) : + self.sess = sess self.data = data self.natoms = self.data.get_natoms() self.ntypes = self.data.get_ntypes() @@ -147,17 +148,17 @@ def comp_v_dw (self, -class TestSmooth(Inter, unittest.TestCase): +class TestSmooth(Inter, tf.test.TestCase): # def __init__ (self, *args, **kwargs): # data = Data() # Inter.__init__(self, data) - # unittest.TestCase.__init__(self, *args, **kwargs) + # tf.test.TestCase.__init__(self, *args, **kwargs) # self.controller = object() def setUp(self): self.places = 5 data = Data() - Inter.setUp(self, data) + Inter.setUp(self, data, sess=self.test_session().__enter__()) def test_force (self) : force_test(self, self, suffix = '_smth') @@ -172,13 +173,13 @@ def test_virial_dw (self) : virial_dw_test(self, self, suffix = '_smth') -class TestSeAPbc(unittest.TestCase): +class TestSeAPbc(tf.test.TestCase): def test_pbc(self): data = Data() inter0 = Inter() inter1 = Inter() - inter0.setUp(data, pbc = True) - inter1.setUp(data, pbc = False) + inter0.setUp(data, pbc = True, sess=self.test_session().__enter__()) + inter1.setUp(data, pbc = False, sess=self.test_session().__enter__()) inter0.net_w_i = np.copy(np.ones(inter0.ndescrpt)) inter1.net_w_i = np.copy(np.ones(inter1.ndescrpt)) @@ -218,8 +219,8 @@ def test_pbc_small_box(self): data1 = Data(box_scale = 2) inter0 = Inter() inter1 = Inter() - inter0.setUp(data0, pbc = True) - inter1.setUp(data1, pbc = False) + inter0.setUp(data0, pbc = True, sess=self.test_session().__enter__()) + inter1.setUp(data1, pbc = False, sess=self.test_session().__enter__()) inter0.net_w_i = np.copy(np.ones(inter0.ndescrpt)) inter1.net_w_i = np.copy(np.ones(inter1.ndescrpt)) diff --git a/source/tests/test_dipole_se_a.py b/source/tests/test_dipole_se_a.py index 26a86c137b..3687ba47f2 100644 --- a/source/tests/test_dipole_se_a.py +++ b/source/tests/test_dipole_se_a.py @@ -14,7 +14,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -90,7 +90,7 @@ def test_model(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [p, gp] = sess.run([dipole, gdipole], feed_dict = feed_dict_test) diff --git a/source/tests/test_embedding_net.py b/source/tests/test_embedding_net.py index 4c07ea575e..03084a8267 100644 --- a/source/tests/test_embedding_net.py +++ b/source/tests/test_embedding_net.py @@ -11,9 +11,9 @@ from deepmd.env import GLOBAL_NP_FLOAT_PRECISION from deepmd.env import GLOBAL_ENER_FLOAT_PRECISION -class Inter(unittest.TestCase): +class Inter(tf.test.TestCase): def setUp (self) : - self.sess = tf.Session() + self.sess = self.test_session().__enter__() self.inputs = tf.constant([ 0., 1., 2.], dtype = tf.float64) self.ndata = 3 self.inputs = tf.reshape(self.inputs, [-1, 1]) diff --git a/source/tests/test_ewald.py b/source/tests/test_ewald.py index 83c5bc3a6d..f4913db36f 100644 --- a/source/tests/test_ewald.py +++ b/source/tests/test_ewald.py @@ -19,7 +19,7 @@ global_default_places = 5 -class TestEwaldRecp (unittest.TestCase) : +class TestEwaldRecp (tf.test.TestCase) : def setUp(self): boxl = 4.5 # NOTICE grid should not change before and after box pert... box_pert = 0.2 @@ -62,7 +62,7 @@ def setUp(self): def test_py_interface(self): hh = 1e-4 places = 4 - sess = tf.Session() + sess = self.test_session().__enter__() t_energy, t_force, t_virial \ = op_module.ewald_recp(self.coord, self.charge, self.nloc, self.box, ewald_h = self.ewald_h, @@ -96,7 +96,7 @@ def test_py_interface(self): def test_force(self): hh = 1e-4 places = 6 - sess = tf.Session() + sess = self.test_session().__enter__() t_energy, t_force, t_virial \ = op_module.ewald_recp(self.coord, self.charge, self.nloc, self.box, ewald_h = self.ewald_h, @@ -138,7 +138,7 @@ def test_force(self): def test_virial(self): hh = 1e-4 places = 6 - sess = tf.Session() + sess = self.test_session().__enter__() t_energy, t_force, t_virial \ = op_module.ewald_recp(self.coord, self.charge, self.nloc, self.box, ewald_h = self.ewald_h, diff --git a/source/tests/test_fitting_ener_type.py b/source/tests/test_fitting_ener_type.py index 71312188a3..25b85391e2 100644 --- a/source/tests/test_fitting_ener_type.py +++ b/source/tests/test_fitting_ener_type.py @@ -15,7 +15,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -97,7 +97,7 @@ def test_fitting(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [pred_atom_ener] = sess.run([atom_ener], feed_dict = feed_dict_test) diff --git a/source/tests/test_model_loc_frame.py b/source/tests/test_model_loc_frame.py index fe4d128c08..3ffce1bf41 100644 --- a/source/tests/test_model_loc_frame.py +++ b/source/tests/test_model_loc_frame.py @@ -14,7 +14,7 @@ GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -96,7 +96,7 @@ def test_model(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [e, f, v] = sess.run([energy, force, virial], feed_dict = feed_dict_test) diff --git a/source/tests/test_model_se_a.py b/source/tests/test_model_se_a.py index 700986e308..e8b12c8553 100644 --- a/source/tests/test_model_se_a.py +++ b/source/tests/test_model_se_a.py @@ -14,7 +14,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -102,7 +102,7 @@ def test_model_atom_ener(self): t_mesh: test_data['default_mesh'], is_training: False } - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [e, f, v] = sess.run([energy, force, virial], feed_dict = feed_dict_test) @@ -189,7 +189,7 @@ def test_model(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [e, f, v] = sess.run([energy, force, virial], feed_dict = feed_dict_test) diff --git a/source/tests/test_model_se_a_aparam.py b/source/tests/test_model_se_a_aparam.py index 77acb2143f..9c9847c242 100644 --- a/source/tests/test_model_se_a_aparam.py +++ b/source/tests/test_model_se_a_aparam.py @@ -13,7 +13,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -95,7 +95,7 @@ def test_model(self): t_aparam: np.reshape(test_data['aparam'] [:numb_test, :], [-1]), is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [e, f, v] = sess.run([energy, force, virial], feed_dict = feed_dict_test) diff --git a/source/tests/test_model_se_a_fparam.py b/source/tests/test_model_se_a_fparam.py index b69ee713e6..23b0da569b 100644 --- a/source/tests/test_model_se_a_fparam.py +++ b/source/tests/test_model_se_a_fparam.py @@ -13,7 +13,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -96,7 +96,7 @@ def test_model(self): t_fparam: np.reshape(test_data['fparam'] [:numb_test, :], [-1]), is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [e, f, v] = sess.run([energy, force, virial], feed_dict = feed_dict_test) diff --git a/source/tests/test_model_se_a_srtab.py b/source/tests/test_model_se_a_srtab.py index d04f011acd..4b87fc0bb4 100644 --- a/source/tests/test_model_se_a_srtab.py +++ b/source/tests/test_model_se_a_srtab.py @@ -23,7 +23,7 @@ def _make_tab(ntype) : prt = np.reshape(prt, [ninter+1, -1]) np.savetxt('tab.xvg', prt.T) -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() _make_tab(2) @@ -117,7 +117,7 @@ def test_model(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [e, f, v] = sess.run([energy, force, virial], feed_dict = feed_dict_test) diff --git a/source/tests/test_model_se_a_type.py b/source/tests/test_model_se_a_type.py index 393cb4c86e..11979108b4 100644 --- a/source/tests/test_model_se_a_type.py +++ b/source/tests/test_model_se_a_type.py @@ -15,7 +15,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -99,7 +99,7 @@ def test_model(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [e, f, v] = sess.run([energy, force, virial], feed_dict = feed_dict_test) diff --git a/source/tests/test_model_se_r.py b/source/tests/test_model_se_r.py index 5172746cc5..56052db8d3 100644 --- a/source/tests/test_model_se_r.py +++ b/source/tests/test_model_se_r.py @@ -13,7 +13,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -91,7 +91,7 @@ def test_model(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [e, f, v] = sess.run([energy, force, virial], feed_dict = feed_dict_test) diff --git a/source/tests/test_model_se_t.py b/source/tests/test_model_se_t.py index 78b06ea7ab..48fffd76b0 100644 --- a/source/tests/test_model_se_t.py +++ b/source/tests/test_model_se_t.py @@ -13,7 +13,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -89,7 +89,7 @@ def test_model(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [e, f, v] = sess.run([energy, force, virial], feed_dict = feed_dict_test) diff --git a/source/tests/test_polar_se_a.py b/source/tests/test_polar_se_a.py index c84bc0aa6d..78da154864 100644 --- a/source/tests/test_polar_se_a.py +++ b/source/tests/test_polar_se_a.py @@ -14,7 +14,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -89,7 +89,7 @@ def test_model(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [p, gp] = sess.run([polar, gpolar], feed_dict = feed_dict_test) diff --git a/source/tests/test_prod_env_mat.py b/source/tests/test_prod_env_mat.py index 2e8aa168f8..4b60da39d9 100644 --- a/source/tests/test_prod_env_mat.py +++ b/source/tests/test_prod_env_mat.py @@ -9,9 +9,9 @@ from deepmd.env import GLOBAL_NP_FLOAT_PRECISION from deepmd.env import GLOBAL_ENER_FLOAT_PRECISION -class TestProdEnvMat(unittest.TestCase): +class TestProdEnvMat(tf.test.TestCase): def setUp(self): - self.sess = tf.Session() + self.sess = self.test_session().__enter__() self.nframes = 2 self.dcoord = [ 12.83, 2.56, 2.18, diff --git a/source/tests/test_prod_force.py b/source/tests/test_prod_force.py index 012fd7ee9b..4135ee0e2e 100644 --- a/source/tests/test_prod_force.py +++ b/source/tests/test_prod_force.py @@ -9,9 +9,9 @@ from deepmd.env import GLOBAL_NP_FLOAT_PRECISION from deepmd.env import GLOBAL_ENER_FLOAT_PRECISION -class TestProdForce(unittest.TestCase): +class TestProdForce(tf.test.TestCase): def setUp(self): - self.sess = tf.Session() + self.sess = self.test_session().__enter__() self.nframes = 2 self.dcoord = [ 12.83, 2.56, 2.18, diff --git a/source/tests/test_prod_force_grad.py b/source/tests/test_prod_force_grad.py index da05406477..8b8ad74ed6 100644 --- a/source/tests/test_prod_force_grad.py +++ b/source/tests/test_prod_force_grad.py @@ -9,9 +9,9 @@ from deepmd.env import GLOBAL_NP_FLOAT_PRECISION from deepmd.env import GLOBAL_ENER_FLOAT_PRECISION -class TestProdForceGrad(unittest.TestCase): +class TestProdForceGrad(tf.test.TestCase): def setUp(self): - self.sess = tf.Session() + self.sess = self.test_session().__enter__() self.nframes = 2 self.dcoord = [ 12.83, 2.56, 2.18, diff --git a/source/tests/test_prod_virial.py b/source/tests/test_prod_virial.py index 4a22e95839..2a11f0acfd 100644 --- a/source/tests/test_prod_virial.py +++ b/source/tests/test_prod_virial.py @@ -9,9 +9,9 @@ from deepmd.env import GLOBAL_NP_FLOAT_PRECISION from deepmd.env import GLOBAL_ENER_FLOAT_PRECISION -class TestProdVirial(unittest.TestCase): +class TestProdVirial(tf.test.TestCase): def setUp(self): - self.sess = tf.Session() + self.sess = self.test_session().__enter__() self.nframes = 2 self.dcoord = [ 12.83, 2.56, 2.18, diff --git a/source/tests/test_prod_virial_grad.py b/source/tests/test_prod_virial_grad.py index 60fdd7f66d..2ca9071a56 100644 --- a/source/tests/test_prod_virial_grad.py +++ b/source/tests/test_prod_virial_grad.py @@ -9,9 +9,9 @@ from deepmd.env import GLOBAL_NP_FLOAT_PRECISION from deepmd.env import GLOBAL_ENER_FLOAT_PRECISION -class TestProdVirialGrad(unittest.TestCase): +class TestProdVirialGrad(tf.test.TestCase): def setUp(self): - self.sess = tf.Session() + self.sess = self.test_session().__enter__() self.nframes = 2 self.dcoord = [ 12.83, 2.56, 2.18, diff --git a/source/tests/test_tab_nonsmth.py b/source/tests/test_tab_nonsmth.py index f6496bbb1d..0d8305cef4 100644 --- a/source/tests/test_tab_nonsmth.py +++ b/source/tests/test_tab_nonsmth.py @@ -31,9 +31,10 @@ def _make_tab(ntype) : class IntplInter(Inter): def setUp (self, - data) : + data, + sess=None) : # tabulated - Inter.setUp(self, data) + Inter.setUp(self, data, sess=sess) _make_tab(data.get_ntypes()) self.srtab = PairTab('tab.xvg') self.smin_alpha = 0.3 @@ -151,7 +152,7 @@ def comp_interpl_ef (self, -class TestTabNonSmooth(IntplInter, unittest.TestCase): +class TestTabNonSmooth(IntplInter, tf.test.TestCase): # def __init__ (self, *args, **kwargs): # self.places = 5 # data = Data() @@ -162,7 +163,7 @@ class TestTabNonSmooth(IntplInter, unittest.TestCase): def setUp(self): self.places = 5 data = Data() - IntplInter.setUp(self, data) + IntplInter.setUp(self, data, sess=self.test_session().__enter__()) def test_force (self) : force_test(self, self, places=5, suffix = '_tab') diff --git a/source/tests/test_tab_smooth.py b/source/tests/test_tab_smooth.py index 55c16d57cc..ab4dc65c7a 100644 --- a/source/tests/test_tab_smooth.py +++ b/source/tests/test_tab_smooth.py @@ -31,9 +31,10 @@ def _make_tab(ntype) : class IntplInter(Inter): def setUp (self, - data) : + data, + sess=None) : # tabulated - Inter.setUp(self, data) + Inter.setUp(self, data, sess=sess) _make_tab(data.get_ntypes()) self.srtab = PairTab('tab.xvg') self.smin_alpha = 0.3 @@ -149,7 +150,7 @@ def comp_ef (self, -class TestTabSmooth(IntplInter, unittest.TestCase): +class TestTabSmooth(IntplInter, tf.test.TestCase): # def __init__ (self, *args, **kwargs): # self.places = 5 # data = Data() @@ -160,7 +161,7 @@ class TestTabSmooth(IntplInter, unittest.TestCase): def setUp(self): self.places = 5 data = Data() - IntplInter.setUp(self, data) + IntplInter.setUp(self, data, sess=self.test_session().__enter__()) def test_force (self) : force_test(self, self, places=5, suffix = '_tab_smth') diff --git a/source/tests/test_type_embed.py b/source/tests/test_type_embed.py index 5f71d8ed93..42d1c86063 100644 --- a/source/tests/test_type_embed.py +++ b/source/tests/test_type_embed.py @@ -3,7 +3,7 @@ from deepmd.env import tf from deepmd.utils.type_embed import embed_atom_type, TypeEmbedNet -class TestTypeEbd(unittest.TestCase): +class TestTypeEbd(tf.test.TestCase): def test_embed_atom_type(self): ntypes = 3 natoms = tf.constant([5, 5, 3, 0, 2]) @@ -19,7 +19,7 @@ def test_embed_atom_type(self): [7,7,7], [7,7,7]] atom_embed = embed_atom_type(ntypes, natoms, type_embedding) - sess = tf.Session() + sess = self.test_session().__enter__() atom_embed = sess.run(atom_embed) for ii in range(5): for jj in range(3): @@ -29,7 +29,7 @@ def test_embed_atom_type(self): def test_type_embed_net(self): ten = TypeEmbedNet([2, 4, 8], seed = 1, uniform_seed = True) type_embedding = ten.build(2) - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) type_embedding = sess.run(type_embedding) diff --git a/source/tests/test_wfc.py b/source/tests/test_wfc.py index c3759429f6..97af7486bd 100644 --- a/source/tests/test_wfc.py +++ b/source/tests/test_wfc.py @@ -13,7 +13,7 @@ GLOBAL_TF_FLOAT_PRECISION = tf.float64 GLOBAL_NP_FLOAT_PRECISION = np.float64 -class TestModel(unittest.TestCase): +class TestModel(tf.test.TestCase): def setUp(self) : gen_data() @@ -83,7 +83,7 @@ def test_model(self): t_mesh: test_data['default_mesh'], is_training: False} - sess = tf.Session() + sess = self.test_session().__enter__() sess.run(tf.global_variables_initializer()) [p] = sess.run([wfc], feed_dict = feed_dict_test)