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()

```

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,

{"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"}]}