bluecast.tests.test_check_gpu_support

Module Contents

Functions

test_check_gpu_support_cpu_fallback()

Test that the function falls back to CPU when no GPU is detected.

test_check_gpu_support_gpu_available()

Test that the function detects GPU support correctly.

test_check_gpu_support_gpu_warning()

Test that the function falls back to CPU if GPU-related warnings are captured.

bluecast.tests.test_check_gpu_support.test_check_gpu_support_cpu_fallback()

Test that the function falls back to CPU when no GPU is detected.

bluecast.tests.test_check_gpu_support.test_check_gpu_support_gpu_available()

Test that the function detects GPU support correctly.

bluecast.tests.test_check_gpu_support.test_check_gpu_support_gpu_warning()

Test that the function falls back to CPU if GPU-related warnings are captured.