update gpu default argument test

waltsims · Oct 27, 2024 · 65e1aa5 · 65e1aa5
1 parent 89cce57
commit 65e1aa5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_simulation_execution_options.py b/tests/test_simulation_execution_options.py
@@ -21,9 +21,10 @@ def test_default_initialization(self):
             self.assertEqual(options.verbose_level, 0)
 
     def test_gpu_simulation_initialization(self):
-        options = SimulationExecutionOptions(is_gpu_simulation=True)
-        self.assertTrue(options.is_gpu_simulation)
-        self.assertEqual(options.binary_name, "kspaceFirstOrder-CUDA")
+        with patch("kwave.options.simulation_execution_options.PLATFORM", "linux"):
+            options = SimulationExecutionOptions(is_gpu_simulation=True)
+            self.assertTrue(options.is_gpu_simulation)
+            self.assertEqual(options.binary_name, "kspaceFirstOrder-CUDA")
 
     def test_binary_name_extension_on_windows(self):
         with patch("kwave.options.simulation_execution_options.PLATFORM", "windows"):