perform a test driven development to read file_name from an ascii fil…

…e in python

exercises-python/template/ex01_basics/main.py

@@ -1,9 +1,17 @@
 #!/usr/bin/env python3
 
+def read_ascii_file(param):
+	try:
+		f = open(param, 'r')
+		return f.read()
+	except FileNotFoundError:
+		print("Oops! That was not a valid file. Try again...")
 
 def main():
-	pass
+	read_ascii_file("file.txt")
 
 
 if __name__ == "__main__":
 	main()
+
+

exercises-python/template/ex01_basics/test_wordcount.py

@@ -0,0 +1,20 @@
+import unittest
+from unittest.mock import mock_open, patch
+from main import *
+
+
+class WordCountTest(unittest.TestCase):
+
+    def setUp(self):
+        # Create a temporary ASCII file for testing
+        self.filename = 'file.txt'
+
+    @patch('builtins.open', new_callable=mock_open, read_data='This is a test data')
+    def test_get_file_name_from_arg(self, mock_file):
+        content = read_ascii_file(self.filename)
+        expected_content = 'This is a test data'
+        self.assertEqual(content, expected_content)
+        mock_file.assert_called_once_with(self.filename, 'r')
+
+
+