1 files changed, 26 insertions, 29 deletions
diff --git a/tests/extmod/machine1.py b/tests/extmod/machine1.py
index 90e6d1717..2f1c0681f 100644
--- a/tests/extmod/machine1.py
+++ b/tests/extmod/machine1.py
@@ -8,39 +8,36 @@ except:
     print("SKIP")
     raise SystemExit
 
-print(machine.mem8)
+import unittest
 
-try:
-    machine.mem16[1]
-except ValueError:
-    print("ValueError")
 
-try:
-    machine.mem16[1] = 1
-except ValueError:
-    print("ValueError")
+class Test(unittest.TestCase):
+    def test_mem8_print(self):
+        self.assertEqual(repr(machine.mem8), "<8-bit memory>")
 
-try:
-    del machine.mem8[0]
-except TypeError:
-    print("TypeError")
+    def test_alignment(self):
+        with self.assertRaises(ValueError):
+            machine.mem16[1]
 
-try:
-    machine.mem8[0:1]
-except TypeError:
-    print("TypeError")
+        with self.assertRaises(ValueError):
+            machine.mem16[1] = 1
 
-try:
-    machine.mem8[0:1] = 10
-except TypeError:
-    print("TypeError")
+    def test_operations(self):
+        with self.assertRaises(TypeError):
+            del machine.mem8[0]
 
-try:
-    machine.mem8["hello"]
-except TypeError:
-    print("TypeError")
+        with self.assertRaises(TypeError):
+            machine.mem8[0:1]
 
-try:
-    machine.mem8["hello"] = 10
-except TypeError:
-    print("TypeError")
+        with self.assertRaises(TypeError):
+            machine.mem8[0:1] = 10
+
+        with self.assertRaises(TypeError):
+            machine.mem8["hello"]
+
+        with self.assertRaises(TypeError):
+            machine.mem8["hello"] = 10
+
+
+if __name__ == "__main__":
+    unittest.main()