file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")

if(APPLE OR WIN32 OR NOT WITH_MKL)
    list(REMOVE_ITEM TEST_OPS test_calibration)
endif()

foreach(src ${TEST_OPS})
    if(src MATCHES "test_calibration")
        py_test(${src} SRCS ${src}.py ENVS FLAGS_use_mkldnn=true)
    else()
        py_test(${src} SRCS ${src}.py)
    endif()
endforeach()