Add a script to run the tests.

PiperOrigin-RevId: 183998520

README.md

@@ -180,6 +180,15 @@ environment. You can see the list of known maps by running:
 $ python -m pysc2.bin.map_list
 ```
 
+## Run the tests
+
+If you want to submit a pull request, please make sure the tests pass on both
+python 2 and 3.
+
+```shell
+$ python -m pysc2.bin.run_tests
+```
+
 # Environment Details
 
 For a full description of the specifics of how the environment is configured,

pysc2/__init__.py

@@ -11,3 +11,14 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+"""PySC2 module: https://github.com/deepmind/pysc2 ."""
+
+import os
+
+
+def load_tests(loader, standard_tests, unused_pattern):
+  """Our tests end in `_test.py`, so need to override the test discovery."""
+  this_dir = os.path.dirname(__file__)
+  package_tests = loader.discover(start_dir=this_dir, pattern="*_test.py")
+  standard_tests.addTests(package_tests)
+  return standard_tests

pysc2/bin/run_tests.py

@@ -0,0 +1,30 @@
+#!/usr/bin/python
+# Copyright 2017 Google Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS-IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Find and run the tests.
+
+Run as: python -m pysc2.bin.run_tests
+"""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
+from absl.testing import absltest
+
+import pysc2
+
+
+if __name__ == '__main__':
+  absltest.main(module=pysc2)