tests(utils): refactor TestPCDFromDir

adosar · Jan 16, 2025 · 757370c · 757370c
1 parent 62943f6
commit 757370c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -92,13 +92,12 @@ def setUp(self):
         self.tempdir = tempfile.TemporaryDirectory(dir='/tmp')
         self.outname = os.path.join(self.tempdir.name, 'pcd_data')
         self.dirname = 'tests/structures'
-        self.names = [Path(i).stem for i in os.listdir(self.dirname)]
 
     def test_pcd_from_dir(self):
         pcd_from_dir(dirname=self.dirname, outname=self.outname)
 
         # Check the number of converted files.
-        self.assertEqual(len(self.names), len(os.listdir(self.outname)))
+        self.assertEqual(len(os.listdir(self.dirname)), len(os.listdir(self.outname)))
 
     def tearDown(self):
         self.tempdir.cleanup()