Cope with webob internal change

toscawidgets · Jun 14, 2010 · 6c83fa0 · 6c83fa0
1 parent 010784c
commit 6c83fa0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_widgets.py b/tests/test_widgets.py
@@ -1,11 +1,16 @@
 from tw2.forms.widgets import *
-from webob import Request, NestedMultiDict
+from webob import Request
 from tw2.core.testbase import assert_in_xml, assert_eq_xml, WidgetTest
 from nose.tools import raises
 from cStringIO import StringIO
 from tw2.core import EmptyField, IntValidator, ValidationError
 from cgi import FieldStorage
 
+import webob
+if hasattr(webob, 'NestedMultiDict'):
+    from webob import NestedMultiDict
+else:
+    from webob.multidict import NestedMultiDict
 
 class TestInputField(WidgetTest):
     widget = InputField