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))