neingeist/check_kernel
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Provide a __str__ method

Browse source
This commit is contained in:
Mike Gerber 2015-05-24 20:33:35 +02:00
parent 790cfe7751
commit c3ecdeb767
2 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
check_kernel
View file

@ -12,6 +12,9 @@ class Version(object):
def __init__(self, version_str): def __init__(self, version_str):
self.version_str = version_str self.version_str = version_str
def __str__(self):
return self.version_str
def components(self): def components(self):
return re.split('\.|-', self.version_str) return re.split('\.|-', self.version_str)

3
check_kernel_test.py
View file

@ -40,6 +40,9 @@ class RunningKernelVersionTestCase(unittest.TestCase):
class VersionTestCase(unittest.TestCase): class VersionTestCase(unittest.TestCase):
def testStr(self):
self.assertEqual(str(Version('1.0')), '1.0')
def testComparingTrivial(self): def testComparingTrivial(self):
self.assertEqual(Version('1.0'), Version('1.0')) self.assertEqual(Version('1.0'), Version('1.0'))
self.assertGreater(Version('2.0'), Version('1.0')) self.assertGreater(Version('2.0'), Version('1.0'))