Use wheel helper function instead of pre-created wheel file

Author: chrahunt

tests/data/packages/headers.dist-0.1-py2.py3-none-any.whl

@@ -148,21 +148,27 @@ def test_basic_install_from_unicode_wheel(script, data):
 
 def get_header_scheme_path_for_script(script, dist_name):
     command = textwrap.dedent(
-        """
+        '''
         from pip._internal.locations import get_scheme
 
         print(get_scheme({!r}).headers)
-        """
+        '''
     ).format(dist_name)
     result = script.run('python', '-c', command).stdout
     return Path(result.strip())
 
 
-def test_install_from_wheel_with_headers(script, data):
+def test_install_from_wheel_with_headers(script):
     """
     Test installing from a wheel file with headers
     """
-    package = data.packages.joinpath("headers.dist-0.1-py2.py3-none-any.whl")
+    package = make_wheel(
+        'headers.dist',
+        '0.1',
+        extra_data_files={
+            'headers/header.h': '',
+        },
+    ).save_to_dir(script.scratch_path)
     result = script.pip('install', package, '--no-index')
     dist_info_folder = script.site_packages / 'headers.dist-0.1.dist-info'
     result.did_create(dist_info_folder)