From da745ae565e43a44fdd8cbd0dc89d9a88d80bad1 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Wed, 29 May 2024 21:00:41 +0200
Subject: [PATCH] revert printing literals with casts
---
src/pystencils/types/types.py | 9 +++------
tests/nbackend/test_extensions.py | 6 +++---
2 files changed, 6 insertions(+), 9 deletions(-)
diff --git a/src/pystencils/types/types.py b/src/pystencils/types/types.py
index ae0a8829d..e7f1ce208 100644
--- a/src/pystencils/types/types.py
+++ b/src/pystencils/types/types.py
@@ -483,7 +483,7 @@ class PsIntegerType(PsScalarType, ABC):
if not isinstance(value, np_dtype):
raise PsTypeError(f"Given value {value} is not of required type {np_dtype}")
unsigned_suffix = "" if self.signed else "u"
- return f"(({self._c_type_without_const()}) {value}{unsigned_suffix})"
+ return f"{value}{unsigned_suffix}"
def create_constant(self, value: Any) -> Any:
np_type = self.NUMPY_TYPES[self._width]
@@ -498,12 +498,9 @@ class PsIntegerType(PsScalarType, ABC):
raise PsTypeError(f"Could not interpret {value} as {repr(self)}")
- def _c_type_without_const(self) -> str:
- prefix = "" if self._signed else "u"
- return f"{prefix}int{self._width}_t"
-
def c_string(self) -> str:
- return f"{self._const_string()}{self._c_type_without_const()}"
+ prefix = "" if self._signed else "u"
+ return f"{self._const_string()}{prefix}int{self._width}_t"
def __repr__(self) -> str:
return f"PsIntegerType( width={self.width}, signed={self.signed}, const={self.const} )"
diff --git a/tests/nbackend/test_extensions.py b/tests/nbackend/test_extensions.py
index 8d600ef76..75726a351 100644
--- a/tests/nbackend/test_extensions.py
+++ b/tests/nbackend/test_extensions.py
@@ -54,6 +54,6 @@ def test_literals():
print(code)
assert "const double x = C;" in code
- assert "CELLS[((int64_t) 0)]" in code
- assert "CELLS[((int64_t) 1)]" in code
- assert "CELLS[((int64_t) 2)]" in code
+ assert "CELLS[0]" in code
+ assert "CELLS[1]" in code
+ assert "CELLS[2]" in code
--
GitLab