from __future__ import absolute_import

from ..composition import SeekableSource

import numpy as np


class IntegerIdentitySource(SeekableSource):
    """Return the integer used as position argument."""

    def __init__(self, size=np.iinfo(np.uint32).max, **kwargs):
        self.parallel_possible = True
        self.cached = True
        self._shape = 1,
        self._size = size
        super(IntegerIdentitySource, self).__init__(name=u"IntegerIdentitySource", **kwargs)

    def _get_data_at(self, position):
        return np.array([position])

    @property
    def dtype(self):
        return np.uint32

from __future__ import absolute_import

from ..composition import SeekableSource

import numpy as np


class IntegerIdentitySource(SeekableSource):
    """Return the integer used as position argument."""

    def __init__(self, size=np.iinfo(np.uint32).max, **kwargs):
        self.parallel_possible = True
        self.cached = True
        self._shape = 1,
        self._dtype = np.uint32
        self._size = size
        super(IntegerIdentitySource, self).__init__(name=u"IntegerIdentitySource", **kwargs)

    def _get_data_at(self, position):
        return np.array([position], dtype=self._dtype)

    @property
    def dtype(self):
        return self._dtype

Set identity integer source data type
Set identity integer source data type

  from __future__ import absolute_import
  
  from ..composition import SeekableSource
  
  import numpy as np
  
  
  class IntegerIdentitySource(SeekableSource):
      """Return the integer used as position argument."""
  
      def __init__(self, size=np.iinfo(np.uint32).max, **kwargs):
          self.parallel_possible = True
          self.cached = True
          self._shape = 1,
+         self._dtype = np.uint32
          self._size = size
          super(IntegerIdentitySource, self).__init__(name=u"IntegerIdentitySource", **kwargs)
  
      def _get_data_at(self, position):
-         return np.array([position])
+         return np.array([position], dtype=self._dtype)
  
      @property
      def dtype(self):
-         return np.uint32
+         return self._dtype
  
Set identity integer source data type
## Code Before:
from __future__ import absolute_import

from ..composition import SeekableSource

import numpy as np


class IntegerIdentitySource(SeekableSource):
    """Return the integer used as position argument."""

    def __init__(self, size=np.iinfo(np.uint32).max, **kwargs):
        self.parallel_possible = True
        self.cached = True
        self._shape = 1,
        self._size = size
        super(IntegerIdentitySource, self).__init__(name=u"IntegerIdentitySource", **kwargs)

    def _get_data_at(self, position):
        return np.array([position])

    @property
    def dtype(self):
        return np.uint32

## Instruction:
Set identity integer source data type
## Code After:
from __future__ import absolute_import

from ..composition import SeekableSource

import numpy as np


class IntegerIdentitySource(SeekableSource):
    """Return the integer used as position argument."""

    def __init__(self, size=np.iinfo(np.uint32).max, **kwargs):
        self.parallel_possible = True
        self.cached = True
        self._shape = 1,
        self._dtype = np.uint32
        self._size = size
        super(IntegerIdentitySource, self).__init__(name=u"IntegerIdentitySource", **kwargs)

    def _get_data_at(self, position):
        return np.array([position], dtype=self._dtype)

    @property
    def dtype(self):
        return self._dtype
