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 (
99 from_array_extended ,
@@ -107,6 +107,7 @@ def test_simple_big(self):
107107 got = ref .run (None , feeds )
108108 self .assertEqualArray (expected , got [0 ])
109109
110+ @skipif_ci_apple ("libomp is missing" )
110111 def test_constant_folding (self ):
111112 with contextlib .redirect_stdout (io .StringIO ()):
112113 g = GraphBuilder (verbose = 10 )
@@ -133,6 +134,7 @@ def test_constant_folding(self):
133134 got = ref .run (None , feeds )
134135 self .assertEqualArray (expected , got [0 ])
135136
137+ @skipif_ci_apple ("libomp is missing" )
136138 def test_constant_folding2 (self ):
137139 g = GraphBuilder (
138140 optimization_options = OptimizationOptions (constant_folding = True )
@@ -270,6 +272,7 @@ def test_remove_unused_nodes_simple(self):
270272 got = ref .run (None , feeds )
271273 self .assertEqualArray (expected , got [0 ])
272274
275+ @skipif_ci_apple ("libomp is missing" )
273276 def test_constant_array (self ):
274277 with contextlib .redirect_stdout (io .StringIO ()):
275278 g = GraphBuilder (verbose = 10 )
@@ -290,6 +293,7 @@ def test_constant_array(self):
290293 got = ref .run (None , feeds )
291294 self .assertEqualArray (expected , got [0 ])
292295
296+ @skipif_ci_apple ("libomp is missing" )
293297 def test_constant_array_2 (self ):
294298 with contextlib .redirect_stdout (io .StringIO ()):
295299 g = GraphBuilder (verbose = 10 )
0 commit comments