diff --git a/tests/test_alternative_wrappers.py b/tests/test_alternative_wrappers.py index 3a1b7ec72ab2b874ba073d5800fbc2825efb8178..afbf7af184d2d9165a3516c13bcf8da83f238563 100644 --- a/tests/test_alternative_wrappers.py +++ b/tests/test_alternative_wrappers.py @@ -24,4 +24,10 @@ def test_wrap_walberla(): def test_wrap_tensorflow(): - pass + import pytest + pytest.importorskip("pystencils_autodiff") + + from pystencils_autodiff.backends.astnodes import TensorflowModule + + generate_shared_object(tempfile.TemporaryDirectory, None, show_code=True, + framework_module_class=TensorflowModule, generate_code_only=True)