diff --git a/tests/fields/fields.py b/tests/fields/fields.py index ae4575ad..75d698c7 100644 --- a/tests/fields/fields.py +++ b/tests/fields/fields.py @@ -3076,8 +3076,10 @@ class FieldTest(unittest.TestCase): class Doc(Document): foo = StringField(db_field='f') - with self.assertRaises(FieldDoesNotExist): + def test(): Doc(bar='test') + self.assertRaises(FieldDoesNotExist, test) + if __name__ == '__main__': unittest.main()