Enable testing with travis-ci

- renamed strategies/branch/tests/test_core.py
- pytest support in setup.py
- travis badge in README.md

Author: skirpichev

.travis.yml

@@ -0,0 +1,13 @@
+language: python
+python:
+  - 2.6
+  - 2.7
+  - 3.2
+  - 3.3
+  - 3.4
+  - "pypy"
+  - "pypy3"
+script:
+  - python setup.py test
+notifications:
+  email: false

README.md

@@ -1,6 +1,8 @@
 Strategies
 ==========
 
+[![](https://travis-ci.org/logpy/strategies.png)](https://travis-ci.org/logpy/strategies)
+
 A Python library for control flow programming.
 
 This project contains a set of complementary higher order functions that

setup.py

@@ -1,5 +1,28 @@
-from os.path import exists
+import os
+import sys
 from setuptools import setup
+from setuptools.command.test import test as TestCommand
+
+
+class PyTest(TestCommand):
+    def finalize_options(self):
+        TestCommand.finalize_options(self)
+        _test_args = [
+            '--ignore=setup.py',
+            '--verbose',
+            '--durations=5',
+            '--doctest-modules',
+        ]
+        extra_args = os.environ.get('PYTEST_EXTRA_ARGS')
+        if extra_args is not None:
+            _test_args.extend(extra_args.split())
+        self.test_args = _test_args
+        self.test_suite = True
+
+    def run_tests(self):
+        import pytest
+        errno = pytest.main(self.test_args)
+        sys.exit(errno)
 
 setup(name='strategies',
       version='0.2.0',
@@ -8,7 +31,9 @@
       author='Matthew Rocklin',
       author_email='mrocklin@gmail.com',
       install_requires=open('dependencies.txt').read().split('\n'),
+      tests_require=['pytest'],
+      cmdclass = {'test': PyTest},
       license='BSD',
       packages=['strategies', 'strategies.branch'],
-      long_description=open('README.md').read() if exists("README.md") else "",
+      long_description=open('README.md').read() if os.path.exists("README.md") else "",
       zip_safe=False)