test.py

import unittest
import os
import shutil
from unittest.mock import patch
from laz2las import convert_laz_to_las

class TestLazToLas(unittest.TestCase):
    def setUp(self):
        self.test_data_dir = "test_data"
        self.output_dir = "test_output"

        # Create test data directory
        os.makedirs(self.test_data_dir, exist_ok=True)
        os.makedirs(self.output_dir, exist_ok=True)

    def tearDown(self):
        # Clean up test data
        shutil.rmtree(self.test_data_dir)
        shutil.rmtree(self.output_dir)

    def test_convert_laz_to_las_success(self):
        # Create a mock LAZ file
        # ... (replace with actual LAZ file creation logic)

        # Call the function to convert
        result = convert_laz_to_las(laz_file_path, self.output_dir)

        # Assert that the conversion was successful
        self.assertTrue(result)
        self.assertTrue(os.path.exists(output_file))

    def test_convert_laz_to_las_failure(self):
        # Create a mock invalid LAZ file
        # ... (replace with actual invalid LAZ file creation logic)

        # Call the function to convert
        result = convert_laz_to_las(invalid_laz_file_path, self.output_dir)

        # Assert that the conversion failed
        self.assertFalse(result)

    def test_output_dir_creation(self):
        # Test if output directory is created if it doesn't exist
        output_dir = "non_existent_dir"
        result = convert_laz_to_las(laz_file_path, output_dir)

        # Assert that the output directory was created
        self.assertTrue(os.path.exists(output_dir))

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