
from cern.pymad import tfs
import unittest


class TestTFS(unittest.TestCase):
    def test_wrongpath(self):
        with self.assertRaises(ValueError):
            tfs('wrong_file_path')


if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(TestTFS)
    unittest.TextTestRunner(verbosity=2).run(suite)

@@ -115,13 +115,8 @@
     
-with 
 self
@@ -143,27 +143,15 @@
 rror
-):%0A           
+,
  tfs
-(
+, 
 'wro

test was incorrectly written