support mac dylib

samskalicky · Mar 4, 2020 · 981900a · 981900a
1 parent 5cffa74
commit 981900a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/python/unittest/test_extensions.py b/tests/python/unittest/test_extensions.py
@@ -36,7 +36,11 @@ def check_platform():
 def test_custom_op():
     # possible places to find library file
     if (os.name=='posix'):
-        lib = 'libcustomop_lib.so'
+        # check if mac
+        if platform.system() == 'Darwin':
+            lib = 'libcustomop_lib.dylib'
+        else:
+            lib = 'libcustomop_lib.so'
         if os.path.exists(lib):
             fname = lib
         elif os.path.exists(os.path.join(base_path,'build/'+lib)):