prompt: We are currently solving the following issue within our repository. Here is the issue text:
  --- BEGIN ISSUE ---
  Add tests to package
  In `setup.py`, make sure `tests` subdirectories are included in the package (otherwise, it is not possible for a user with an installed version of dask to verify its integrity).
  
  
  --- END ISSUE ---
  
  Below are some code segments, each from a relevant file. One or more of these files may contain bugs.
  --- BEGIN FILES ---
  Path: `setup.py`
  Content:
  ```
  1 #!/usr/bin/env python
  2 
  3 from os.path import exists
  4 from setuptools import setup
  5 import dask
  6 
  7 extras_require = {
  8   'array': ['numpy', 'toolz >= 0.7.2'],
  9   'bag': ['cloudpickle >= 0.2.1', 'toolz >= 0.7.2', 'partd >= 0.3.3'],
  10   'dataframe': ['numpy', 'pandas >= 0.18.0', 'toolz >= 0.7.2',
  11                 'partd >= 0.3.3', 'cloudpickle >= 0.2.1'],
  12   'distributed': ['distributed >= 1.9'],
  13   'imperative': ['toolz >= 0.7.2'],
  14 }
  15 extras_require['complete'] = sorted(set(sum(extras_require.values(), [])))
  16 
  17 setup(name='dask',
  18       version=dask.__version__,
  19       description='Minimal task scheduling abstraction',
  20       url='http://github.com/dask/dask/',
  21       maintainer='Matthew Rocklin',
  22       maintainer_email='mrocklin@gmail.com',
  23       license='BSD',
  24       keywords='task-scheduling parallelism',
  25       packages=['dask', 'dask.array', 'dask.bag', 'dask.store', 'dask.bytes',
  26                 'dask.dataframe', 'dask.dataframe.tseries', 'dask.diagnostics'],
  27       long_description=(open('README.rst').read() if exists('README.rst')
  28                         else ''),
  29       extras_require=extras_require,
  30       zip_safe=False)
  31
  ```
  --- END FILES ---
  
  Please first localize the bug based on the issue statement, and then generate a patch according to the `git diff` format fenced by three backticks.
  
  Here is an example:
  
  ```diff
  diff --git a/examples/server_async.py b/examples/server_async.py
  --- a/examples/server_async.py
  +++ b/examples/server_async.py
  @@ -313,4 +313,4 @@
   
   
   if __name__ == "__main__":
  -    asyncio.run(run_async_server("."), debug=True)
  +    asyncio.run(run_async_server(), debug=True)
  diff --git a/examples/server_sync.py b/examples/server_sync.py
  --- a/examples/server_sync.py
  +++ b/examples/server_sync.py
  @@ -313,5 +313,5 @@
   
   
   if __name__ == "__main__":
  -    server = run_sync_server(".")
  +    server = run_sync_server()
       server.shutdown()
  
  ```
  
golden_diff: diff --git a/setup.py b/setup.py
  --- a/setup.py
  +++ b/setup.py
  @@ -14,6 +14,12 @@
   }
   extras_require['complete'] = sorted(set(sum(extras_require.values(), [])))
   
  +packages = ['dask', 'dask.array', 'dask.bag', 'dask.store', 'dask.bytes',
  +            'dask.dataframe', 'dask.dataframe.tseries', 'dask.diagnostics']
  +
  +tests = [p + '.tests' for p in packages]
  +
  +
   setup(name='dask',
         version=dask.__version__,
         description='Minimal task scheduling abstraction',
  @@ -22,8 +28,7 @@
         maintainer_email='mrocklin@gmail.com',
         license='BSD',
         keywords='task-scheduling parallelism',
  -      packages=['dask', 'dask.array', 'dask.bag', 'dask.store', 'dask.bytes',
  -                'dask.dataframe', 'dask.dataframe.tseries', 'dask.diagnostics'],
  +      packages=packages + tests,
         long_description=(open('README.rst').read() if exists('README.rst')
                           else ''),
         extras_require=extras_require,
  
verification_info: {"golden_diff": "diff --git a/setup.py b/setup.py\n--- a/setup.py\n+++ b/setup.py\n@@ -14,6 +14,12 @@\n }\n extras_require['complete'] = sorted(set(sum(extras_require.values(), [])))\n \n+packages = ['dask', 'dask.array', 'dask.bag', 'dask.store', 'dask.bytes',\n+            'dask.dataframe', 'dask.dataframe.tseries', 'dask.diagnostics']\n+\n+tests = [p + '.tests' for p in packages]\n+\n+\n setup(name='dask',\n       version=dask.__version__,\n       description='Minimal task scheduling abstraction',\n@@ -22,8 +28,7 @@\n       maintainer_email='mrocklin@gmail.com',\n       license='BSD',\n       keywords='task-scheduling parallelism',\n-      packages=['dask', 'dask.array', 'dask.bag', 'dask.store', 'dask.bytes',\n-                'dask.dataframe', 'dask.dataframe.tseries', 'dask.diagnostics'],\n+      packages=packages + tests,\n       long_description=(open('README.rst').read() if exists('README.rst')\n                         else ''),\n       extras_require=extras_require,\n", "issue": "Add tests to package\nIn `setup.py`, make sure `tests` subdirectories are included in the package (otherwise, it is not possible for a user with an installed version of dask to verify its integrity).\n\n", "before_files": [{"content": "#!/usr/bin/env python\n\nfrom os.path import exists\nfrom setuptools import setup\nimport dask\n\nextras_require = {\n  'array': ['numpy', 'toolz >= 0.7.2'],\n  'bag': ['cloudpickle >= 0.2.1', 'toolz >= 0.7.2', 'partd >= 0.3.3'],\n  'dataframe': ['numpy', 'pandas >= 0.18.0', 'toolz >= 0.7.2',\n                'partd >= 0.3.3', 'cloudpickle >= 0.2.1'],\n  'distributed': ['distributed >= 1.9'],\n  'imperative': ['toolz >= 0.7.2'],\n}\nextras_require['complete'] = sorted(set(sum(extras_require.values(), [])))\n\nsetup(name='dask',\n      version=dask.__version__,\n      description='Minimal task scheduling abstraction',\n      url='http://github.com/dask/dask/',\n      maintainer='Matthew Rocklin',\n      maintainer_email='mrocklin@gmail.com',\n      license='BSD',\n      keywords='task-scheduling parallelism',\n      packages=['dask', 'dask.array', 'dask.bag', 'dask.store', 'dask.bytes',\n                'dask.dataframe', 'dask.dataframe.tseries', 'dask.diagnostics'],\n      long_description=(open('README.rst').read() if exists('README.rst')\n                        else ''),\n      extras_require=extras_require,\n      zip_safe=False)\n", "path": "setup.py"}], "after_files": [{"content": "#!/usr/bin/env python\n\nfrom os.path import exists\nfrom setuptools import setup\nimport dask\n\nextras_require = {\n  'array': ['numpy', 'toolz >= 0.7.2'],\n  'bag': ['cloudpickle >= 0.2.1', 'toolz >= 0.7.2', 'partd >= 0.3.3'],\n  'dataframe': ['numpy', 'pandas >= 0.18.0', 'toolz >= 0.7.2',\n                'partd >= 0.3.3', 'cloudpickle >= 0.2.1'],\n  'distributed': ['distributed >= 1.9'],\n  'imperative': ['toolz >= 0.7.2'],\n}\nextras_require['complete'] = sorted(set(sum(extras_require.values(), [])))\n\npackages = ['dask', 'dask.array', 'dask.bag', 'dask.store', 'dask.bytes',\n            'dask.dataframe', 'dask.dataframe.tseries', 'dask.diagnostics']\n\ntests = [p + '.tests' for p in packages]\n\n\nsetup(name='dask',\n      version=dask.__version__,\n      description='Minimal task scheduling abstraction',\n      url='http://github.com/dask/dask/',\n      maintainer='Matthew Rocklin',\n      maintainer_email='mrocklin@gmail.com',\n      license='BSD',\n      keywords='task-scheduling parallelism',\n      packages=packages + tests,\n      long_description=(open('README.rst').read() if exists('README.rst')\n                        else ''),\n      extras_require=extras_require,\n      zip_safe=False)\n", "path": "setup.py"}]}