33import unittest
44import numpy as np
55import onnx
6- from onnx_array_api .ext_test_case import ExtTestCase
6+ from onnx_array_api .ext_test_case import ExtTestCase , skipif_ci_apple
77from onnx_array_api .graph_api .graph_builder import GraphBuilder ,OptimizationOptions
88from onnx_array_api .reference import (
99from_array_extended ,
@@ -107,7 +107,7 @@ def test_simple_big(self):
107107got = ref .run (None ,feeds )
108108self .assertEqualArray (expected ,got [0 ])
109109
110- @skip_ci_apple ("libomp is missing" )
110+ @skipif_ci_apple ("libomp is missing" )
111111def test_constant_folding (self ):
112112with contextlib .redirect_stdout (io .StringIO ()):
113113g = GraphBuilder (verbose = 10 )
@@ -134,7 +134,7 @@ def test_constant_folding(self):
134134got = ref .run (None ,feeds )
135135self .assertEqualArray (expected ,got [0 ])
136136
137- @skip_ci_apple ("libomp is missing" )
137+ @skipif_ci_apple ("libomp is missing" )
138138def test_constant_folding2 (self ):
139139g = GraphBuilder (
140140optimization_options = OptimizationOptions (constant_folding = True )
@@ -272,7 +272,7 @@ def test_remove_unused_nodes_simple(self):
272272got = ref .run (None ,feeds )
273273self .assertEqualArray (expected ,got [0 ])
274274
275- @skip_ci_apple ("libomp is missing" )
275+ @skipif_ci_apple ("libomp is missing" )
276276def test_constant_array (self ):
277277with contextlib .redirect_stdout (io .StringIO ()):
278278g = GraphBuilder (verbose = 10 )
@@ -293,7 +293,7 @@ def test_constant_array(self):
293293got = ref .run (None ,feeds )
294294self .assertEqualArray (expected ,got [0 ])
295295
296- @skip_ci_apple ("libomp is missing" )
296+ @skipif_ci_apple ("libomp is missing" )
297297def test_constant_array_2 (self ):
298298with contextlib .redirect_stdout (io .StringIO ()):
299299g = GraphBuilder (verbose = 10 )