def test_save_end_no_tag_end_of_module(self): class_line, end_line = self.container._find_class_end('ClassNoEndTagEndOfModule') self.assertNotEqual(class_line, -1) self.assertNotEqual(end_line, -1) self.assertEqual(len(self.container.lines) - 1, end_line) self.assertTrue(self.container.save_module('output1.py')) end_token = create_end_class_token('ClassNoEndTagEndOfModule') with open('output1.py') as f: lines = f.readlines() self.assertTrue(end_token in lines[len(lines)-1])
def test_save_end_no_tag(self): class_line, end_line = self.container._find_class_end('ClassNoEndTag') self.assertNotEqual(class_line, -1) self.assertNotEqual(end_line, -1) self.assertTrue(self.container.save_module('output2.py')) end_token = create_end_class_token('ClassNoEndTag') with open('output2.py') as f: lines = f.readlines() # for index in xrange(end_line-1, end_line+2): # print '[%d] %s' % (index, lines[index].strip()) self.assertTrue(end_token in lines[end_line], 'Line: %s' % lines[end_line])
def test_create_end_class_token(self): token = create_end_class_token(class_name='AClass') self.assertTrue('/AClass' in token)