input: Write a python function to get discography of an artist from Wikipedia.
  assert get_discography("The Beatles") is not None
  assert get_discography("Taylor Swift") is not None
  assert get_discography("Adele") is not None
output: import requests
  from bs4 import BeautifulSoup
  
  def get_discography(artist_name):
      search_url = f"https://en.wikipedia.org/wiki/{artist_name.replace(' ', '_')}_discography"
      response = requests.get(search_url)
      if response.status_code != 200:
          return None
  
      soup = BeautifulSoup(response.content, 'html.parser')
      discography = []
  
      for heading in soup.find_all(['h2', 'h3']):
          if 'Discography' in heading.text:
              for sibling in heading.find_next_siblings():
                  if sibling.name == 'h2':
                      break
                  if sibling.name == 'ul':
                      for li in sibling.find_all('li'):
                          album = li.get_text()
                          discography.append(album)
              break
      
      return discography if discography else None