def stripComments(code: str) -> str: """Remove comments from a Python code string.""" lines = code.split("\n") lines = [line.split("#")[0].rstrip() for line in lines] lines = [line for line in lines if line] lines = [f"{line}\n" for line in lines] return "".join(lines) print("Testing stripComments()...", end="") code = """\ # here's a comment! def foo(x): return x + 1 # here's another one """ result = """\ def foo(x): return x + 1 """ assert stripComments(code) == result code = """\ def g(x): # Here are some comments # which must be removed # by stripComments return x * 7 """ result = """\ def g(x): return x * 7 """ assert stripComments(code) == result code = """\ def doIHaveAnyComments(): return 'No' """ result = """\ def doIHaveAnyComments(): return 'No' """ assert stripComments(code) == result code = """\ def f(x): #This function returns x + 5 return x + 5 """ result = """\ def f(x): return x + 5 """ assert stripComments(code) == result assert stripComments("") == "" print("Passed!")