1 files changed, 6 insertions, 1 deletions
diff --git a/test/__init__.py b/test/__init__.py
index cf5f36a..46f9f81 100644
--- a/test/__init__.py
+++ b/test/__init__.py
@@ -1,7 +1,7 @@
 from mako.template import Template
 import unittest
 import os
-from mako.compat import py3k, py26
+from mako.compat import py3k, py26, py33
 from mako import compat
 from mako.util import update_wrapper
 import re
@@ -62,6 +62,11 @@ def teardown():
     import shutil
     shutil.rmtree(module_base, True)
 
+if py33:
+    from unittest import mock
+else:
+    import mock
+
 @contextlib.contextmanager
 def raises(except_cls, message=None):
     try: