Use base class for testing

NWChemEx · Dec 12, 2023 · ae0cfdf · ae0cfdf
1 parent d5e0e2d
commit ae0cfdf
Showing 1 changed file with 10 additions and 18 deletions.
diff --git a/tests/python/unit_tests/derivative/test_derivative.py b/tests/python/unit_tests/derivative/test_derivative.py
@@ -16,22 +16,14 @@
 import pluginplay
 import unittest
 
-class TestEnergy(unittest.TestCase):
-    def test_aoenergynucleargradient_pt(self):
-        pt = simde.AOEnergyNuclearGradient()
-        self.assertIn('AOs', pt.inputs())
-        self.assertIn('Chemical System', pt.inputs())
-        self.assertIn('Arg 1', pt.inputs())
-        self.assertEqual(len(pt.inputs()), 3)
-        self.assertIn('Derivative', pt.results())
-        self.assertEqual(len(pt.results()), 1)
+class TestAOEnergyNuclearGradient(unittest.TestCase):
+    def setUp(self):
+        self.pt = simde.AOEnergyNuclearGradient()
+        self.input_labels = ['AOs', 'Chemical System', 'Arg 1']
+        self.return_labels = ['Derivative']
 
-    def test_aoenergynuclearhessian_pt(self):
-        pt = simde.AOEnergyNuclearHessian()
-        self.assertIn('AOs', pt.inputs())
-        self.assertIn('Chemical System', pt.inputs())
-        self.assertIn('Arg 1', pt.inputs())
-        self.assertIn('Arg 2', pt.inputs())
-        self.assertEqual(len(pt.inputs()), 4)
-        self.assertIn('Derivative', pt.results())
-        self.assertEqual(len(pt.results()), 1)
+class TestAOEnergyNuclearHessian(unittest.TestCase):
+    def setUp(self):
+        self.pt = simde.AOEnergyNuclearHessian()
+        self.input_labels = ['AOs', 'Chemical System', 'Arg 1', 'Arg 2']
+        self.return_labels = ['Derivative']