Added validation for frames in TableLayer

This commit is contained in:
O . O 2019-04-04 11:47:59 -04:00
Родитель 0bb30fc0bb
Коммит 018107738d
1 изменённых файлов: 6 добавлений и 3 удалений

Просмотреть файл

@ -280,10 +280,13 @@ class TableLayer(HasTraits):
# zAxisReverse
def __init__(self, parent=None, table=None, frame=None, **kwargs):
# TODO: need to validate reference frame
self.table = table
self.frame = frame
# Validate frame
if frame.lower() not in VALID_FRAMES:
raise ValueError('frame should be one of {0}'.format('/'.join(sorted(str(x) for x in VALID_FRAMES))))
self.frame = frame.capitalize()
self.parent = parent
self.id = str(uuid.uuid4())