Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
tomotools
Nabu
Commits
734fe1c0
Commit
734fe1c0
authored
Jan 06, 2022
by
Pierre Paleo
Browse files
Add test_sino_array_subtraction_cuda
parent
811c42d3
Changes
1
Hide whitespace changes
Inline
Side-by-side
nabu/reconstruction/tests/test_sino_normalization.py
View file @
734fe1c0
...
...
@@ -64,3 +64,25 @@ class TestSinoNormalization:
compare_normalizations
(
array_1D
)
compare_normalizations
(
array_2D
)
@
pytest
.
mark
.
skipif
(
not
(
__has_pycuda__
),
reason
=
"Need pycuda for sinogram normalization with cuda backend"
)
def
test_sino_array_subtraction_cuda
(
self
):
with
pytest
.
raises
(
ValueError
):
CudaSinoNormalization
(
kind
=
"subtract_array"
,
sinos_shape
=
self
.
sino
.
shape
)
array_1D
=
np
.
arange
(
self
.
sino
.
shape
[
-
1
])
array_2D
=
np
.
arange
(
self
.
sino
.
size
).
reshape
(
self
.
sino
.
shape
)
def
compare_normalizations
(
normalization_arr
):
sino_normalization
=
CudaSinoNormalization
(
kind
=
"subtract_array"
,
sinos_shape
=
self
.
sino
.
shape
,
normalization_array
=
normalization_arr
)
sino
=
garray
.
to_gpu
(
self
.
sino
)
sino_normalization
.
normalize
(
sino
)
ref
=
self
.
sino
-
normalization_arr
assert
np
.
allclose
(
sino
.
get
(),
ref
)
compare_normalizations
(
array_1D
)
compare_normalizations
(
array_2D
)
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment