update demo-assignments

Author: rambasnet

demo-assignments/A0-OOP/hello/tests/test_helloworld.py

@@ -12,7 +12,7 @@
 import unittest
 from unittest.mock import patch
 from io import StringIO
-from hello import HelloWorld
+from helloworld import HelloWorld
 
 
 class TestHelloWorld(unittest.TestCase):
@@ -21,27 +21,18 @@ class TestHelloWorld(unittest.TestCase):
     """
 
     def setUp(self) -> None:
-        """
-        Setup method
-        :return: None
+        """Setup method
         """
         self.hello = HelloWorld()
 
     @patch('sys.stdout', new_callable=StringIO)
     def test_print_message(self, mock_stdout: StringIO) -> None:
-        """
-        Tests printMessage method
-        :return: None
+        """Tests printMessage method
         """
         self.hello.print_message()
         self.assertEqual(mock_stdout.getvalue(), 'Hello World!\n')
 
-    def test_getMessage(self) -> None:
-        """
-        Tests getMessage method
-        :return: None
-        """
-        self.assertEqual(self.hello.get_message(), 'Hello World!')
-
     def test_message(self) -> None:
+        """Test message property
+        """
         self.assertEqual(self.hello.message, 'Hello World!')