From 9277ca553600eb0672e5103440f019d2c1a3fcc9 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 27 Feb 2020 11:41:52 +0100
Subject: [PATCH] Add test for Tensorflow wrapping
---
tests/test_alternative_wrappers.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/tests/test_alternative_wrappers.py b/tests/test_alternative_wrappers.py
index 3a1b7ec..afbf7af 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)
--
GitLab