Skip to content
Snippets Groups Projects
Commit 4bae07a3 authored by Michael Kuron's avatar Michael Kuron :mortar_board:
Browse files

flake8 complex numbers

parent da296e88
Branches
Tags
No related merge requests found
...@@ -270,7 +270,8 @@ if( PyErr_Occurred() ) {{ return NULL; }} ...@@ -270,7 +270,8 @@ if( PyErr_Occurred() ) {{ return NULL; }}
template_extract_complex = """ template_extract_complex = """
PyObject * obj_{name} = PyDict_GetItemString(kwargs, "{name}"); PyObject * obj_{name} = PyDict_GetItemString(kwargs, "{name}");
if( obj_{name} == NULL) {{ PyErr_SetString(PyExc_TypeError, "Keyword argument '{name}' missing"); return NULL; }}; if( obj_{name} == NULL) {{ PyErr_SetString(PyExc_TypeError, "Keyword argument '{name}' missing"); return NULL; }};
{target_type} {name}{{ ({real_type}) {extract_function_real}( obj_{name} ), ({real_type}) {extract_function_imag}( obj_{name} ) }}; {target_type} {name}{{ ({real_type}) {extract_function_real}( obj_{name} ),
({real_type}) {extract_function_imag}( obj_{name} ) }};
if( PyErr_Occurred() ) {{ return NULL; }} if( PyErr_Occurred() ) {{ return NULL; }}
""" """
...@@ -409,7 +410,8 @@ def create_function_boilerplate_code(parameter_info, name, insert_checks=True): ...@@ -409,7 +410,8 @@ def create_function_boilerplate_code(parameter_info, name, insert_checks=True):
pre_call_code += template_extract_complex.format(extract_function_real=extract_function[0], pre_call_code += template_extract_complex.format(extract_function_real=extract_function[0],
extract_function_imag=extract_function[1], extract_function_imag=extract_function[1],
target_type=target_type, target_type=target_type,
real_type="float" if target_type == "ComplexFloat" else "double", real_type="float" if target_type == "ComplexFloat"
else "double",
name=param.symbol.name) name=param.symbol.name)
else: else:
pre_call_code += template_extract_scalar.format(extract_function=extract_function, pre_call_code += template_extract_scalar.format(extract_function=extract_function,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment