version : modified __repr__ to return a string representation of the object + unit tests

This commit is contained in:
alalazo 2016-07-06 15:24:10 +02:00
parent d687e332ad
commit 32e086f44a
2 changed files with 9 additions and 1 deletions

View File

@ -389,3 +389,11 @@ def test_formatted_strings(self):
self.assertEqual(v.dotted, '1.2.3')
self.assertEqual(v.dashed, '1-2-3')
self.assertEqual(v.underscored, '1_2_3')
def test_repr_and_str(self):
a = Version('1.2.3')
self.assertEqual(repr(a), 'Version(\'1.2.3\')')
b = eval(repr(a))
self.assertEqual(a, b)
self.assertEqual(str(a), '1.2.3')
self.assertEqual(str(a), str(b))

View File

@ -197,7 +197,7 @@ def __getitem__(self, idx):
return tuple(self.version[idx])
def __repr__(self):
return self.string
return 'Version(' + repr(self.string) + ')'
def __str__(self):
return self.string