From 201f3008b1c451c93d112e5905fe7e64259a8d7e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bastien=20G=C3=A9rard?= Date: Sun, 17 Feb 2019 22:56:56 +0100 Subject: [PATCH] Fix for py3 in binary field test --- tests/fields/test_binary_field.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/tests/fields/test_binary_field.py b/tests/fields/test_binary_field.py index b0427766..8af75d4e 100644 --- a/tests/fields/test_binary_field.py +++ b/tests/fields/test_binary_field.py @@ -119,7 +119,10 @@ class TestBinaryField(MongoDBTestCase): set__bin_field=BIN_VALUE ) self.assertEqual(doc.some_field, 'test') - self.assertEqual(doc.bin_field, Binary(BIN_VALUE)) + if six.PY3: + self.assertEqual(doc.bin_field, BIN_VALUE) + else: + self.assertEqual(doc.bin_field, Binary(BIN_VALUE)) def test_update_one(self): """Ensures no regression of bug #1127""" @@ -134,4 +137,7 @@ class TestBinaryField(MongoDBTestCase): n_updated = MyDocument.objects(bin_field=bin_data).update_one(bin_field=BIN_VALUE) self.assertEqual(n_updated, 1) fetched = MyDocument.objects.with_id(doc.id) - self.assertEqual(fetched.bin_field, Binary(BIN_VALUE)) + if six.PY3: + self.assertEqual(fetched.bin_field, BIN_VALUE) + else: + self.assertEqual(fetched.bin_field, Binary(BIN_VALUE))