Support ./setup.py test

Author: methane

setup.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function
 
@@ -18,5 +19,7 @@
     url='https://github.com/KLab/Flask-PyMemcache',
     description="pymemcache integration for Flask",
     long_description=readme,
-    install_requires=["Flask", "pymemcache>=1.2.3", "six"],
+    install_requires=["Flask", "pymemcache>=1.2.3"],
+    test_require=['pytest'],
+    test_suite='test_flask_pymemcache',
 )

test_flask_pymemcache.py

@@ -1,22 +1,24 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function
+from unittest import TestCase
 
 import pymemcache.client
 import flask
 import flask.ext.pymemcache
 
 
-def test_simple():
-    pymc = pymemcache.client.Client(('localhost', 11211))
-    memcache = flask.ext.pymemcache.FlaskPyMemcache()
-    app = flask.Flask(__name__)
-    app.config['PYMEMCACHE'] = {
-        'server': ('localhost', 11211),
-        'key_prefix': b'px'}
-    memcache.init_app(app)
+class TestFlaskPyMemcache(TestCase):
+    def test_simple(self):
+        pymc = pymemcache.client.Client(('localhost', 11211))
+        memcache = flask.ext.pymemcache.FlaskPyMemcache()
+        app = flask.Flask(__name__)
+        app.config['PYMEMCACHE'] = {
+            'server': ('localhost', 11211),
+            'key_prefix': b'px'}
+        memcache.init_app(app)
 
-    with app.app_context():
-        memcache.client.set(b'foo', b'bar')
-        assert memcache.client.get(b'foo') == b'bar'
+        with app.app_context():
+            memcache.client.set(b'foo', b'bar')
+            assert memcache.client.get(b'foo') == b'bar'
 
-    assert pymc.get(b'pxfoo') == b'bar'
+        assert pymc.get(b'pxfoo') == b'bar'