Skip to content

Commit f535c19

Browse files
committed
Remove hardcoded data shape assertion in small pipeline test
The shape of the data being transferred is in principle dependent on the max slices calculation, which in turn is dependent on the GPU model being used. As the GPU model can vary widely, the max slices and therefore the data shape can vary widely, making this assertion difficult to pass if run on different GPUs. As the purpose of the small pipeline tests is not focused on specific data shapes, but functional behaviour of pipeline execution, asserting that there's a line in the logfile corresponding to a data transfer occurring for GPU 0 is sufficient, and there's no need to be specific about the shape of the data that was transferred.
1 parent 3126192 commit f535c19

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/test_pipeline_small.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,7 @@ def test_run_pipeline_FBP3d_tomobar_denoising(
163163
assert "Preview: (0:180, 0:128, 0:160)" in verbose_log_contents
164164
assert "Data shape is (180, 128, 160) of type uint16" in verbose_log_contents
165165
assert "The amount of the available GPU memory is" in verbose_log_contents
166-
assert (
167-
"Using GPU 0 to transfer data of shape (180, 128, 160)" in verbose_log_contents
168-
)
166+
assert "Using GPU 0 to transfer data of shape" in verbose_log_contents
169167

170168

171169
# TODO: rewrite and move to test_pipeline_big

0 commit comments

Comments
 (0)