From ae1ea301709c420e3543c62236f7fbba2d9403b6 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Mon, 16 Dec 2019 11:00:25 +0100
Subject: [PATCH] Also rename KernelFunctionCall -> FunctionCall in test

---
 tests/test_framework_printer.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/tests/test_framework_printer.py b/tests/test_framework_printer.py
index ffab27e..b7dac07 100644
--- a/tests/test_framework_printer.py
+++ b/tests/test_framework_printer.py
@@ -6,12 +6,12 @@
 
 """
 import pytest
+import sympy as sp
 
 import pystencils
-import sympy as sp
 from pystencils.astnodes import Block
 from pystencils_autodiff.framework_integration.astnodes import (
-    DestructuringBindingsForFieldClass, KernelFunctionCall, WrapperFunction, generate_kernel_call)
+    DestructuringBindingsForFieldClass, FunctionCall, WrapperFunction, generate_kernel_call)
 from pystencils_autodiff.framework_integration.printer import FrameworkIntegrationPrinter
 
 # TODO
@@ -31,7 +31,7 @@ def test_pure_call():
 
     for target in ('cpu', 'gpu'):
         ast = pystencils.create_kernel(forward_assignments, target=target)
-        kernel_call_ast = KernelFunctionCall(ast)
+        kernel_call_ast = FunctionCall(ast)
         code = FrameworkIntegrationPrinter()(kernel_call_ast)
         print(code)
 
@@ -44,7 +44,7 @@ def test_call_with_destructuring():
 
     for target in ('cpu', 'gpu'):
         ast = pystencils.create_kernel(forward_assignments, target=target)
-        kernel_call_ast = KernelFunctionCall(ast)
+        kernel_call_ast = FunctionCall(ast)
         wrapper = DestructuringBindingsForFieldClass(kernel_call_ast)
         code = FrameworkIntegrationPrinter()(wrapper)
         print(code)
@@ -59,7 +59,7 @@ def test_call_with_destructuring_fixed_size():
 
     for target in ('cpu', 'gpu'):
         ast = pystencils.create_kernel(forward_assignments, target=target)
-        kernel_call_ast = KernelFunctionCall(ast)
+        kernel_call_ast = FunctionCall(ast)
         wrapper = DestructuringBindingsForFieldClass(kernel_call_ast)
         code = FrameworkIntegrationPrinter()(wrapper)
         print(code)
@@ -73,14 +73,14 @@ def test_wrapper_function():
 
     for target in ('cpu', 'gpu'):
         ast = pystencils.create_kernel(forward_assignments, target=target)
-        kernel_call_ast = KernelFunctionCall(ast)
+        kernel_call_ast = FunctionCall(ast)
         wrapper = WrapperFunction(DestructuringBindingsForFieldClass(kernel_call_ast))
         code = FrameworkIntegrationPrinter()(wrapper)
         print(code)
 
     for target in ('cpu', 'gpu'):
         ast = pystencils.create_kernel(forward_assignments, target=target)
-        kernel_call_ast = KernelFunctionCall(ast)
+        kernel_call_ast = FunctionCall(ast)
         wrapper = WrapperFunction(Block([kernel_call_ast]))
         code = FrameworkIntegrationPrinter()(wrapper)
         print(code)
-- 
GitLab