Also rename KernelFunctionCall -> FunctionCall in test

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)