def test_linecount(self): b = Buffer('') self.assertEqual(1, b.linecount) b = Buffer('Hello World!') self.assertEqual(1, b.linecount) b = Buffer('\n') self.assertEqual(2, b.linecount)
def setUp(self): testfile = os.path.splitext(__file__)[0] + '.py' with open(testfile) as f: self.text = str(f.read()) self.buf = Buffer(self.text, whitespace='')