Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enable testing of cirq_google notebooks. #5766

Merged
merged 14 commits into from
Sep 3, 2022
Merged

Enable testing of cirq_google notebooks. #5766

Changes from 1 commit
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
3e270dd
First draft of testing cirq_google notebooks.
dstrain115 Jul 14, 2022
b82ee23
formatting
dstrain115 Jul 14, 2022
a55cf2f
Fix colab notebook to work in tests.
dstrain115 Jul 14, 2022
8a222a6
Use signed_in instead of hacky colab check.
dstrain115 Jul 14, 2022
600a38d
Ignore a few more notebooks temporarily.
dstrain115 Jul 14, 2022
3c19109
Missing comma
dstrain115 Jul 14, 2022
ed56d1f
Fix coverage in a possibly hacky way.
dstrain115 Jul 14, 2022
fbeeb96
Add back in test.
dstrain115 Jul 15, 2022
6c0f01f
Fix azure-quantum skip test.
dstrain115 Jul 15, 2022
36be3f4
Use mock.patch.dict to mock the google.colab module
pavoljuhas Jul 16, 2022
edc4b1a
Merge branch 'master' into cirq_google_testable
dstrain115 Sep 1, 2022
dda3067
Address sorted review comment.
dstrain115 Sep 3, 2022
d02f74e
Merge branch 'cirq_google_testable' of https://github.com/dstrain115/…
dstrain115 Sep 3, 2022
18ed45d
Merge branch 'master' into cirq_google_testable
dstrain115 Sep 3, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Use mock.patch.dict to mock the google.colab module
  • Loading branch information
@pavoljuhas
pavoljuhas committed Jul 16, 2022
commit 36be3f491b7685e38d7b80adc29300ac4fb42eea
17 changes: 4 additions & 13 deletions cirq-google/cirq_google/engine/qcs_notebook_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

import sys
import unittest.mock as mock
import pytest

Expand Down Expand Up @@ -97,13 +98,9 @@ def test_get_qcs_objects_for_notebook_no_processors(engine_mock):
_ = get_qcs_objects_for_notebook()


@mock.patch.dict('sys.modules', {'google.colab': mock.Mock()})
@mock.patch('cirq_google.engine.qcs_notebook.get_engine')
def test_get_qcs_objects_for_notebook_auth_succeeds(engine_mock):
# force google.colab to import
auth_mock = mock.Mock()
import sys

sys.modules['google.colab'] = auth_mock
fake_processor = cg.engine.SimulatedLocalProcessor(
processor_id='tester', project_name='mock_project', device=cg.Sycamore
)
Expand All @@ -116,16 +113,12 @@ def test_get_qcs_objects_for_notebook_auth_succeeds(engine_mock):
assert result.project_id == 'mock_project'
assert len(result.device.metadata.qubit_set) == 54

del sys.modules['google.colab']


@mock.patch.dict('sys.modules', {'google.colab': mock.Mock()})
@mock.patch('cirq_google.engine.qcs_notebook.get_engine')
def test_get_qcs_objects_for_notebook_auth_fails(engine_mock):
# force google.colab to import
auth_mock = mock.Mock()
import sys
auth_mock = sys.modules['google.colab']

sys.modules['google.colab'] = auth_mock
auth_mock.auth.authenticate_user = mock.Mock(side_effect=Exception('mock auth failure'))
fake_processor = cg.engine.SimulatedLocalProcessor(
processor_id='tester', project_name='mock_project', device=cg.Sycamore
Expand All @@ -139,5 +132,3 @@ def test_get_qcs_objects_for_notebook_auth_fails(engine_mock):
assert not result.signed_in
assert result.is_simulator
assert result.project_id == 'fake_project'

del sys.modules['google.colab']