code: """Checks that dist/* contains the number of wheels built from the
  .github/workflows/wheels.yml config."""
  import yaml
  from pathlib import Path
  import sys
  
  gh_wheel_path = Path.cwd() / ".github" / "workflows" / "wheels.yml"
  with gh_wheel_path.open('r') as f:
      wheel_config = yaml.safe_load(f)
  
  build_matrix = wheel_config['jobs']['build_wheels']['strategy']['matrix']
  n_python_versions = len(build_matrix['python'])
  
  # For each python version we have: 5 wheels
  # 1 osx wheel (x86_64)
  # 2 linux wheel (i686 + x86_64)
  # 2 windows wheel (win32 + wind_amd64)
  n_wheels = 5 * n_python_versions
  
  # plus one more for the sdist
  n_wheels += 1
  
  # aarch64 builds from travis
  travis_config_path = Path.cwd() / ".travis.yml"
  with travis_config_path.open('r') as f:
      travis_config = yaml.safe_load(f)
  
  jobs = travis_config['jobs']['include']
  travis_builds = [j for j in jobs
                   if any("CIBW_BUILD" in env for env in j["env"])]
  n_wheels += len(travis_builds)
  
  dist_files = list(Path("dist").glob('**/*'))
  n_dist_files = len(dist_files)
  
  if n_dist_files != n_wheels:
      print(f"Expected {n_wheels} wheels in dist/* but "
            f"got {n_dist_files} artifacts instead.")
      sys.exit(1)
  
  print(f"dist/* has the expected {n_wheels} wheels:")
  print("\n".join(file.name for file in dist_files))
  