Force repetition counts to be integers

This commit is contained in:
Jan Petykiewicz 2019-12-06 22:28:11 -08:00
parent f3669f2dfd
commit 97b7eda21a

View File

@ -31,8 +31,8 @@ class GridRepetition:
'_mirrored', '_mirrored',
'_a_vector', '_a_vector',
'_b_vector', '_b_vector',
'a_count', '_a_count',
'b_count', '_b_count',
'identifier') 'identifier')
pattern: 'Pattern' pattern: 'Pattern'
@ -45,8 +45,8 @@ class GridRepetition:
_a_vector: numpy.ndarray _a_vector: numpy.ndarray
_b_vector: numpy.ndarray or None _b_vector: numpy.ndarray or None
a_count: int _a_count: int
b_count: int _b_count: int
identifier: Tuple identifier: Tuple
@ -209,6 +209,27 @@ class GridRepetition:
raise PatternError('b_vector must be convertible to size-2 ndarray') raise PatternError('b_vector must be convertible to size-2 ndarray')
self._b_vector = val.flatten() self._b_vector = val.flatten()
# a_count property
@property
def a_count(self) -> int:
return self._a_count
@a_count.setter
def a_count(self, val: int):
if val != int(val):
raise PatternError('a_count must be convertable to an int!')
self._a_count = int(val)
# b_count property
@property
def b_count(self) -> int:
return self._b_count
@b_count.setter
def b_count(self, val: int):
if val != int(val):
raise PatternError('b_count must be convertable to an int!')
self._b_count = int(val)
def as_pattern(self) -> 'Pattern': def as_pattern(self) -> 'Pattern':
""" """