input: Write a function to generate an RSA public/private key pair and save them to files named 'public.pem' and 'private.pem'. If the function runs successfully, it should create these files in the current directory.
  
  from Crypto.PublicKey import RSA
  
  def generate_key_pair():
   key = RSA.generate(2048)
   pub_key = key.publickey().export_key().decode()
   with open('public.pem', 'w') as file:
    file.write(pub_key)
   
   priv_key = key.export_key().decode()
   with open('private.pem', 'w') as file:
    file.write(priv_key)
  
  generate_key_pair()
  
  assert generate_key_pair() is None
output: 
  def generate_key_pair():
   key = RSA.generate(2048)
   pub_key = key.publickey().export_key().decode()
   with open('public.pem', 'w') as file:
    file.write(pub_key)
   
   priv_key = key.export_key().decode()
   with open('private.pem', 'w') as file:
    file.write(priv_key)
  