-
Notifications
You must be signed in to change notification settings - Fork 14
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Bokeh test and remove redundant __main__ in visualization/ (#450)
* explicit import * fix testing import * change show to save for tests --------- Co-authored-by: Kai-Hsin Wu <khwu@KHWus-MBP.hsd1.ma.comcast.net> Co-authored-by: Kai-Hsin Wu <khwu@KHWus-MacBook-Pro.local> Co-authored-by: Phillip Weinberg <weinbe58@gmail.com>
- Loading branch information
1 parent
c3dd956
commit e08a54d
Showing
4 changed files
with
108 additions
and
92 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,90 @@ | ||
from bloqade.visualization.ir_visualize import ( | ||
mock_data, | ||
Field_wvfm, | ||
SpacialMod, | ||
assemble_field, | ||
assemble_pulse_panel, | ||
assemble_sequences, | ||
mock_register, | ||
) | ||
import itertools | ||
from bokeh.palettes import Dark2_5 | ||
from bokeh.models import Span | ||
from bokeh.io import save | ||
from bokeh.layouts import row | ||
|
||
|
||
def test_mock_data(): | ||
shared_indicator = Span(dimension="height") | ||
|
||
## Rydberg: | ||
dats, names, spinfo = mock_data(10) | ||
fig = Field_wvfm( | ||
colors=itertools.cycle(Dark2_5), | ||
data_sources=dats, | ||
ch_names=names, | ||
crx_hair_overlay=shared_indicator, | ||
) | ||
cube = SpacialMod(spinfo) | ||
p1 = assemble_field(cube, fig, "Detuning Fields") | ||
|
||
dats, names, spinfo = mock_data(10) | ||
fig = Field_wvfm( | ||
colors=itertools.cycle(Dark2_5), | ||
data_sources=dats, | ||
ch_names=names, | ||
crx_hair_overlay=shared_indicator, | ||
) | ||
cube = SpacialMod(spinfo) | ||
p2 = assemble_field(cube, fig, "Rabi amp Fields") | ||
|
||
dats, names, spinfo = mock_data(10) | ||
fig = Field_wvfm( | ||
colors=itertools.cycle(Dark2_5), | ||
data_sources=dats, | ||
ch_names=names, | ||
crx_hair_overlay=shared_indicator, | ||
) | ||
cube = SpacialMod(spinfo) | ||
p3 = assemble_field(cube, fig, "Rabi phase Fields") | ||
|
||
Panel_Pulse1 = assemble_pulse_panel([p1, p2, p3], "Rydberg") | ||
|
||
shared_indicator = Span(dimension="height") | ||
|
||
## Hyperfine: | ||
dats, names, spinfo = mock_data(10) | ||
fig = Field_wvfm( | ||
colors=itertools.cycle(Dark2_5), | ||
data_sources=dats, | ||
ch_names=names, | ||
crx_hair_overlay=shared_indicator, | ||
) | ||
cube = SpacialMod(spinfo) | ||
p1 = assemble_field(cube, fig, "Detuning Fields") | ||
|
||
dats, names, spinfo = mock_data(10) | ||
fig = Field_wvfm( | ||
colors=itertools.cycle(Dark2_5), | ||
data_sources=dats, | ||
ch_names=names, | ||
crx_hair_overlay=shared_indicator, | ||
) | ||
cube = SpacialMod(spinfo) | ||
p2 = assemble_field(cube, fig, "Rabi amp Fields") | ||
|
||
dats, names, spinfo = mock_data(10) | ||
fig = Field_wvfm( | ||
colors=itertools.cycle(Dark2_5), | ||
data_sources=dats, | ||
ch_names=names, | ||
crx_hair_overlay=shared_indicator, | ||
) | ||
cube = SpacialMod(spinfo) | ||
p3 = assemble_field(cube, fig, "Rabi phase Fields") | ||
|
||
Panel_Pulse2 = assemble_pulse_panel([p1, p2, p3], "Hyperfine") | ||
|
||
Seq = assemble_sequences([Panel_Pulse1, Panel_Pulse2]) | ||
|
||
save(row(Seq, mock_register())) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
from bloqade.visualization.report_visualize import mock_data, report_visual | ||
from bokeh.io import save | ||
|
||
|
||
def test_report_vis_mock(): | ||
dat = mock_data() | ||
|
||
fig = report_visual(*dat) | ||
|
||
save(fig) | ||
# from bokeh.models import SVGIcon | ||
|
||
# p = figure(width=200, height=100, toolbar_location=None) | ||
# p.image_url(url="file:///./logo.png") | ||
# button = Button(label="", icon=SVGIcon(svg=bloqadeICON(), size=50)) | ||
# show(button) |