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)
