Fix tests

Author: Roy Williams

swagger_py_codegen/flask.py

@@ -96,13 +96,21 @@ def _type(parameters):
 
     return url, params
 
+if six.PY3:
+    def _remove_characters(text, deletechars):
+        return text.translate({ord(x): None for x in deletechars})
+else:
+    def _remove_characters(text, deletechars):
+        return text.translate(None, deletechars)
 
 def _path_to_endpoint(swagger_path):
-    return swagger_path.strip('/').replace('/', '_').replace('-', '_').translate(None, '{}')
+    return _remove_characters(
+        swagger_path.strip('/').replace('/', '_').replace('-', '_'),
+        '{}')
 
 
 def _path_to_resource_name(swagger_path):
-    return swagger_path.title().translate(None, '{}/_-')
+    return _remove_characters(swagger_path.title(), '{}/_-')
 
 
 def _location(swagger_location):

swagger_py_codegen/parser.py

@@ -8,7 +8,7 @@
 
 
 def schema_var_name(path):
-    return ''.join(map(string.capitalize, path))
+    return ''.join(map(str.capitalize, path))
 
 
 class RefNode(dict):

tests/test_flask.py

@@ -1,4 +1,5 @@
 from __future__ import absolute_import
+
 from swagger_py_codegen.parser import Swagger
 from swagger_py_codegen.flask import (
     _swagger_to_flask_url,
@@ -135,6 +136,7 @@ def test_process_data():
     swagger = Swagger(data)
     generator = FlaskGenerator(swagger)
     schemas, routes, view1, view2 = list(generator.generate())[:4]
+    view1, view2 = sorted([view1, view2], key=lambda x: x.data['name'])
     assert ('posts_post_id', 'GET') in schemas.data['validators']
     assert schemas.data['validators'][('posts_post_id', 'GET')]['args']['properties']['page']['type'] == 'integer'
     assert view1.data['url'] == '/posts/<int:post_id>'