From b50083809509fe69028188f8445e9c6a71ee3d6d Mon Sep 17 00:00:00 2001
From: jo2-holdsworth <jack2.holdsworth@live.uwe.ac.uk>
Date: Tue, 7 Mar 2023 15:24:34 +0000
Subject: [PATCH] assert

---
 assert_tests.py | 8 ++++++++
 main.py         | 5 +++--
 morseunit.py    | 8 ++++++++
 3 files changed, 19 insertions(+), 2 deletions(-)
 create mode 100644 assert_tests.py
 create mode 100644 morseunit.py

diff --git a/assert_tests.py b/assert_tests.py
new file mode 100644
index 0000000..5e61e76
--- /dev/null
+++ b/assert_tests.py
@@ -0,0 +1,8 @@
+import morse
+def test_encode_us():
+    print(morse.encode('us'))
+    assert morse.encode('us') == '..- ... ', "Should be ..- ..."
+
+if __name__ == "__main__":
+    test_encode_us()
+    print('Everything passed')
\ No newline at end of file
diff --git a/main.py b/main.py
index 533fa29..9d0aa48 100644
--- a/main.py
+++ b/main.py
@@ -2,6 +2,7 @@ import morse
 
 if __name__ == "__main__":
     e = morse.encode('us')
-    print(e)
+    print('%s' % e)
     d = morse.decode(e)
-    print(d)
+    assert morse.encode('us') == '..- ...', "Should be ..-"
+    assert morse.decode('..- ...') == 'us', "Should be ..-"
diff --git a/morseunit.py b/morseunit.py
new file mode 100644
index 0000000..bd4c3d0
--- /dev/null
+++ b/morseunit.py
@@ -0,0 +1,8 @@
+import unittest
+import morse
+class TestMorse(unittest.TestCase):
+    def test_encode_us(self):
+        self.assertEqual( morse.encode('us'), '..- ... ')
+
+if __name__ == '__main__':
+    unittest.main()
\ No newline at end of file
-- 
GitLab