Python port: Added shebang and main() methods.

diff --git a/ports/python/mhash384.py b/ports/python/mhash384.py
index 2dea175..a6cffc4 100644 (file)
--- a/ports/python/mhash384.py
+++ b/ports/python/mhash384.py
@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+
 """MHash-384 - Simple fast portable secure hashing library
 Copyright(c) 2016-2019 LoRd_MuldeR <mulder2@gmx.de>
 
@@ -19,7 +21,12 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SO
 
 import struct #required for unpack()
 
+# ===========================================================================
+# MHASH-384 CLASS
+# ===========================================================================
+
 class MHash384:
+
        # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
        # CONSTANTS
        # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -892,3 +899,12 @@ class MHash384:
        @classmethod
        def version(cls):
                return cls.__version_major, cls.__version_minor, cls.__version_patch
+
+
+# ===========================================================================
+# MAIN
+# ===========================================================================
+
+if __name__ == '__main__':
+       import mhash384_selftest
+       mhash384_selftest.main()

diff --git a/ports/python/mhash384_selftest.py b/ports/python/mhash384_selftest.py
index 48871ae..65ed3ee 100644 (file)
--- a/ports/python/mhash384_selftest.py
+++ b/ports/python/mhash384_selftest.py
@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+
 """MHash-384 - Simple fast portable secure hashing library
 Copyright(c) 2016-2019 LoRd_MuldeR <mulder2@gmx.de>
 
@@ -19,7 +21,12 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SO
 
 import sys, unittest, mhash384
 
-class TestStringMethods(unittest.TestCase):
+# ===========================================================================
+# SELF-TEST CLASS
+# ===========================================================================
+
+class MHash384Selftest(unittest.TestCase):
+
        # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
        # INITIALIZATION
        # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -98,5 +105,13 @@ class TestStringMethods(unittest.TestCase):
                print('\r{0} - {1}'.format(digest, 'OK' if digest == expected else 'Error!'))
                self.assertEqual(digest, expected)
 
+
+# ===========================================================================
+# MAIN
+# ===========================================================================
+
+def main():
+       unittest.main(module=__name__)
+
 if __name__ == '__main__':
-       unittest.main()
+       main()