def test_lesson_1(self): """Animals that can speak are correctly identified""" self.assertIsInstance( OOLessons.lesson1(ast.parse("")), dict ) self.assertEqual( OOLessons.lesson1(ast.parse("")), { 'success': False, 'errors': [{ 'msg': "You have to use 'import' statement."}]} ) self.assertEqual( OOLessons.lesson1(ast.parse("""import wrong_module""")), { 'success': False, 'errors': [{ 'msg': 'Wrong module.'}]} ) self.assertEqual( OOLessons.lesson1(ast.parse("""\nimport bla\nimport ble\nclass Foo:\tpass""")), { 'success': False, 'errors': [ { 'msg': 'Wrong module.'}]} ) self.assertEqual( OOLessons.lesson1(ast.parse("""import time\n""")), { 'success': True, 'errors': []} ) self.assertEqual( OOLessons.lesson1(ast.parse("""\n\nimport time\n""")), { 'success': True, 'errors': []} ) self.assertEqual( OOLessons.lesson1(ast.parse("""\nimport bla\nimport time\nclass Foo:\tpass""")), { 'success': True, 'errors': []} )