From c3ecdeb767df7d87b85f650c0ae7586cf41ea7a6 Mon Sep 17 00:00:00 2001 From: Mike Gerber Date: Sun, 24 May 2015 20:33:35 +0200 Subject: [PATCH] Provide a __str__ method --- check_kernel | 3 +++ check_kernel_test.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/check_kernel b/check_kernel index ee8052d..10a575e 100755 --- a/check_kernel +++ b/check_kernel @@ -12,6 +12,9 @@ class Version(object): def __init__(self, version_str): self.version_str = version_str + def __str__(self): + return self.version_str + def components(self): return re.split('\.|-', self.version_str) diff --git a/check_kernel_test.py b/check_kernel_test.py index cbe6d8b..c16a4c6 100644 --- a/check_kernel_test.py +++ b/check_kernel_test.py @@ -40,6 +40,9 @@ class RunningKernelVersionTestCase(unittest.TestCase): class VersionTestCase(unittest.TestCase): + def testStr(self): + self.assertEqual(str(Version('1.0')), '1.0') + def testComparingTrivial(self): self.assertEqual(Version('1.0'), Version('1.0')) self.assertGreater(Version('2.0'), Version('1.0'))