From f86951c96f662771b6fdb846ed4ca9c08f4a36b9 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 3 Dec 2020 18:18:44 +0100
Subject: [PATCH] Fix test for old Ubuntu
---
pystencils_tests/test_astnodes.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/pystencils_tests/test_astnodes.py b/pystencils_tests/test_astnodes.py
index 98c755efd..4ae85bd61 100644
--- a/pystencils_tests/test_astnodes.py
+++ b/pystencils_tests/test_astnodes.py
@@ -1,9 +1,9 @@
+import pytest
import sympy as sp
-import pystencils as ps
+import pystencils as ps
from pystencils import Assignment
-from pystencils.astnodes import Block, SkipIteration, LoopOverCoordinate, SympyAssignment
-from sympy.codegen.rewriting import optims_c99
+from pystencils.astnodes import Block, LoopOverCoordinate, SkipIteration, SympyAssignment
dst = ps.fields('dst(8): double[2D]')
s = sp.symbols('s_:8')
@@ -83,7 +83,8 @@ def test_loop_over_coordinate():
def test_sympy_assignment():
-
+ pytest.importorskip('sympy.codegen.rewriting')
+ from sympy.codegen.rewriting import optims_c99
assignment = SympyAssignment(dst[0, 0](0), sp.log(x + 3) / sp.log(2) + sp.log(x ** 2 + 1))
assignment.optimize(optims_c99)
--
GitLab