tests_unit.py

import unittest
from functions.get_files_info import get_files_info

class TestGetFiles(unittest.TestCase):

    def test_1(self):
        result = get_files_info("calculator", ".")
        self.assertEqual(result.split('\n'), [
            "- main.py: file_size=576 bytes, is_dir=False",
            "- pkg: file_size=4096 bytes, is_dir=True",
            "- tests.py: file_size=1343 bytes, is_dir=False"
        ])

    def test_2(self):
        result = get_files_info("calculator", "pkg")
        self.assertEqual(result.split('\n'), [
            "- calculator.py: file_size=1739 bytes, is_dir=False",
            "- render.py: file_size=768 bytes, is_dir=False"
        ])
        
    
    def test_3(self):
        result = get_files_info("calculator", "/bin")
        self.assertEqual(result, 'Error: Cannot list "/bin" as it is outside the permitted working directory')
    
    
    def test_4(self):
        result = get_files_info("calculator", "..")
        self.assertEqual(result, 'Error: Cannot list ".." as it is outside the permitted working directory')

if __name__ == '__main__':
    unittest.main()