Skip to content

Commit

Permalink
Update integration tests (#820)
Browse files Browse the repository at this point in the history
  • Loading branch information
kt474 committed Feb 22, 2024
1 parent 5ce508c commit 6ff92c2
Showing 1 changed file with 6 additions and 7 deletions.
13 changes: 6 additions & 7 deletions test/integration/test_session.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,9 +44,8 @@ def test_session_id(self):
self.assertEqual(backend.session.session_id, None)
self.assertTrue(backend.session.active)
job1 = backend.run(bell())
self.assertEqual(job1._session_id, job1.job_id())
job2 = backend.run(bell())
self.assertFalse(job2._session_id == job2.job_id())
self.assertTrue(job1.result())
self.assertEqual(job1._session_id, None)

def test_backend_run_with_session(self):
"""Test that 'shots' parameter is transferred correctly"""
Expand Down Expand Up @@ -89,11 +88,12 @@ def test_run_after_cancel(self):

backend.open_session()
job2 = backend.run(bell())
self.assertIsNotNone(job2._session_id)
self.assertTrue(job2.result())
backend.cancel_session()

job3 = backend.run(circuits=bell())
self.assertIsNone(backend.session)
self.assertTrue(job3.result())
self.assertIsNone(job3._session_id)

def test_session_as_context_manager(self):
Expand All @@ -104,9 +104,8 @@ def test_session_as_context_manager(self):
with backend.open_session() as session:
job1 = backend.run(bell())
session_id = session.session_id
self.assertEqual(session_id, job1.job_id())
job2 = backend.run(bell())
self.assertFalse(session_id == job2.job_id())
self.assertTrue(job1.result())
self.assertIsNone(session_id)

def test_run_after_cancel_as_context_manager(self):
"""Test run after cancel in context manager"""
Expand Down

0 comments on commit 6ff92c2

Please sign in to comment.