ran unittest2pytest
This commit is contained in:
		| @@ -10,6 +10,7 @@ from mongoengine import * | ||||
| from mongoengine.connection import get_db | ||||
|  | ||||
| from tests.utils import MongoDBTestCase | ||||
| import pytest | ||||
|  | ||||
|  | ||||
| class TestLongField(MongoDBTestCase): | ||||
| @@ -24,10 +25,10 @@ class TestLongField(MongoDBTestCase): | ||||
|  | ||||
|         doc = TestLongFieldConsideredAsInt64(some_long=42).save() | ||||
|         db = get_db() | ||||
|         self.assertIsInstance( | ||||
|         assert isinstance( | ||||
|             db.test_long_field_considered_as_int64.find()[0]["some_long"], Int64 | ||||
|         ) | ||||
|         self.assertIsInstance(doc.some_long, six.integer_types) | ||||
|         assert isinstance(doc.some_long, six.integer_types) | ||||
|  | ||||
|     def test_long_validation(self): | ||||
|         """Ensure that invalid values cannot be assigned to long fields. | ||||
| @@ -41,11 +42,14 @@ class TestLongField(MongoDBTestCase): | ||||
|         doc.validate() | ||||
|  | ||||
|         doc.value = -1 | ||||
|         self.assertRaises(ValidationError, doc.validate) | ||||
|         with pytest.raises(ValidationError): | ||||
|             doc.validate() | ||||
|         doc.value = 120 | ||||
|         self.assertRaises(ValidationError, doc.validate) | ||||
|         with pytest.raises(ValidationError): | ||||
|             doc.validate() | ||||
|         doc.value = "ten" | ||||
|         self.assertRaises(ValidationError, doc.validate) | ||||
|         with pytest.raises(ValidationError): | ||||
|             doc.validate() | ||||
|  | ||||
|     def test_long_ne_operator(self): | ||||
|         class TestDocument(Document): | ||||
| @@ -56,4 +60,4 @@ class TestLongField(MongoDBTestCase): | ||||
|         TestDocument(long_fld=None).save() | ||||
|         TestDocument(long_fld=1).save() | ||||
|  | ||||
|         self.assertEqual(1, TestDocument.objects(long_fld__ne=None).count()) | ||||
|         assert 1 == TestDocument.objects(long_fld__ne=None).count() | ||||
|   | ||||
		Reference in New Issue
	
	Block a user