Python StringIO.tellの例

コード例 #1

ファイル: test_parmed_gromacs_cpp.py プロジェクト: zyguostx/ParmEd

 def test_ifdef(self):
     """ Tests CPreProcessor #ifdef/#else/#endif preprocessing """
     f = StringIO('#ifdef MY_DEFINE\nPPVAR is set to MY_DEFINE\n'
                  '#else\nMY_DEFINE is not set\n#endif\n')
     pp = CPreProcessor(f)  # no defines
     self.assertEqual(pp.read().strip(), 'MY_DEFINE is not set')
     f.seek(0)
     self.assertEqual(f.tell(), 0)
     self.assertEqual(pp.tell(), 0)
     pp = CPreProcessor(f, defines=dict(MY_DEFINE='SUCCESS'))
     self.assertEqual(pp.read().strip(), 'PPVAR is set to SUCCESS')

コード例 #2

ファイル: test_parmed_gromacs_cpp.py プロジェクト: ChayaSt/ParmEd

 def test_ifdef(self):
     """ Tests CPreProcessor #ifdef/#else/#endif preprocessing """
     f = StringIO('#ifdef MY_DEFINE\nPPVAR is set to MY_DEFINE\n'
                  '#else\nMY_DEFINE is not set\n#endif\n')
     pp = CPreProcessor(f) # no defines
     self.assertEqual(pp.read().strip(), 'MY_DEFINE is not set')
     f.seek(0)
     self.assertEqual(f.tell(), 0)
     self.assertEqual(pp.tell(), 0)
     pp = CPreProcessor(f, defines=dict(MY_DEFINE='SUCCESS'))
     self.assertEqual(pp.read().strip(), 'PPVAR is set to SUCCESS')