Skip to content

Commit 6b4947a

Browse files
committed
better assertions for "is"
1 parent e08284b commit 6b4947a

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

test/asynchronous/test_session.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -378,9 +378,9 @@ async def test_cursor_clone(self):
378378

379379
async with self.client.start_session() as s:
380380
cursor = coll.find(session=s)
381-
self.assertTrue(cursor.session is s)
381+
self.assertIs(cursor.session, s)
382382
clone = cursor.clone()
383-
self.assertTrue(clone.session is s)
383+
self.assertIs(clone.session, s)
384384

385385
# No explicit session.
386386
cursor = coll.find(batch_size=2)
@@ -392,7 +392,7 @@ async def test_cursor_clone(self):
392392
await anext(clone)
393393
self.assertIsNone(clone.session)
394394
self.assertIsNotNone(clone._session)
395-
self.assertFalse(cursor._session is clone._session)
395+
self.assertIsNot(cursor._session, clone._session)
396396
await cursor.close()
397397
await clone.close()
398398

test/test_session.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -378,9 +378,9 @@ def test_cursor_clone(self):
378378

379379
with self.client.start_session() as s:
380380
cursor = coll.find(session=s)
381-
self.assertTrue(cursor.session is s)
381+
self.assertIs(cursor.session, s)
382382
clone = cursor.clone()
383-
self.assertTrue(clone.session is s)
383+
self.assertIs(clone.session, s)
384384

385385
# No explicit session.
386386
cursor = coll.find(batch_size=2)
@@ -392,7 +392,7 @@ def test_cursor_clone(self):
392392
next(clone)
393393
self.assertIsNone(clone.session)
394394
self.assertIsNotNone(clone._session)
395-
self.assertFalse(cursor._session is clone._session)
395+
self.assertIsNot(cursor._session, clone._session)
396396
cursor.close()
397397
clone.close()
398398

0 commit comments

Comments
 (0)