def test_isSUFile(self): """ Tests the isSU method. """ # Test all SEG Y files in the test directory. for file in self.files.keys(): file = os.path.join(self.path, file) self.assertEqual(isSU(file), False) # Also check all the other files in the test directory and they should # not work. Just check certain files to ensure reproducibility. files = ['test_core.py', 'test_segy.py', '__init__.py'] for file in files: file = os.path.join(self.dir, file) self.assertEqual(isSU(file), False) # Check an actual Seismic Unix file. file = os.path.join(self.path, '1.su_first_trace') self.assertEqual(isSU(file), True)