Check that the type in Colour and Permissions are int.

This commit is contained in:
Rapptz committed 2017-02-21 08:19:32 -05:00
1 parent b7488d7c6e
commit 702d596af9
2 files changed
+6

No files matched your search

+3
View File
@@ -53,6 +53,9 @@ class Colour:
__slots__ = ('value',)
def __init__(self, value):
if not isinstance(value, int):
raise TypeError('Expected int parameter, received %s instead.' % value.__class__.__name__)
self.value = value
def _get_byte(self, byte):
+3
View File
@@ -69,6 +69,9 @@ class Permissions:
__slots__ = ('value',)
def __init__(self, permissions=0):
if not isinstance(permissions, int):
raise TypeError('Expected int parameter, received %s instead.' % permissions.__class__.__name__)
self.value = permissions
def __eq__(self, other):