Skip to content
Snippets Groups Projects

Update for slice_viz function

Merged ofhkr requested to merge Update_sliceviz into main
7 files
+ 83
40
Compare changes
  • Side-by-side
  • Inline

Files

+ 2
2
@@ -35,7 +35,7 @@ def test_grid_pred():
@@ -35,7 +35,7 @@ def test_grid_pred():
in_targ_pred = qim3d.utils.models.inference(train_set,model)
in_targ_pred = qim3d.utils.models.inference(train_set,model)
fig = qim3d.viz.grid_pred(in_targ_pred)
fig = qim3d.viz.grid_pred(in_targ_pred)
assert (fig.get_figwidth(),fig.get_figheight()) == (2*(n),10)
assert (fig.get_figwidth(),fig.get_figheight()) == (2*(n),10)
temp_data(folder,remove = True)
temp_data(folder,remove = True)
@@ -45,7 +45,7 @@ def test_grid_pred():
@@ -45,7 +45,7 @@ def test_grid_pred():
def test_slice_viz():
def test_slice_viz():
example_volume = ones(10,10,10)
example_volume = ones(10,10,10)
img_width = 3
img_width = 3
fig = qim3d.viz.slice_viz(example_volume,img_width = img_width)
fig = qim3d.viz.slice_viz(example_volume,n_slices = 1, img_width = img_width)
assert fig.get_figwidth() == img_width
assert fig.get_figwidth() == img_width
Loading