def test_get_current_class_returns_false_when_not_in_class(self):
current_buffer = self.build_buffer_helper()
self.assertEqual(False, sut.get_current_class(2, current_buffer))
def test_get_not_in_class_message(self):
current_buffer = self.build_buffer_helper()
self.assertEqual("You don't appear to be in a class", sut.get_current_class("Bad line", current_buffer))
def test_get_valid_class_name(self):
current_line1 = 17
current_line2 = 24
current_buffer = self.build_buffer_helper()
self.assertEqual("Example1", sut.get_current_class(current_line1, current_buffer))
self.assertEqual("Example2", sut.get_current_class(current_line2, current_buffer))
def test_get_valid_class_name(self):
current_line1 = " print('This is a testD')\n"
current_line2 = " print('This is a test4b')\n"
current_buffer = self.build_buffer_helper()
self.assertEqual("Example1", sut.get_current_class(current_line1, current_buffer))
self.assertEqual("Example2", sut.get_current_class(current_line2, current_buffer))
