List all the imports from the python code

Question

def list_imports(lines):
    pass

# DO NOT TOUCH THE BELOW CODE
class TestListImports(unittest.TestCase):

    def test_01(self):
        code_with_imports = "