diff --git a/pystencils/data_types.py b/pystencils/data_types.py index 8bf28e777e9bf9deec30f21cafe8cb3b7ac77b17..e4c6723e250f3bd2c4a2cf47b4b0325599c0ae32 100644 --- a/pystencils/data_types.py +++ b/pystencils/data_types.py @@ -127,6 +127,13 @@ class TypedSymbol(sp.Symbol): def __getnewargs__(self): return self.name, self.dtype + @property + def is_integer(self): + if hasattr(self.dtype, 'numpy_dtype'): + return np.issubdtype(self.dtype.numpy_dtype, np.integer) + else: + return False + def create_type(specification): """Creates a subclass of Type according to a string or an object of subclass Type.