test_modules.py

Go to the documentation of this file.

import os
 
from genmypy import cli
 
from .utils import assert_output_equals, message_path, service_path, temporary_directory
 
 
def test_std_msgs_msg(expected_dir, std_msgs_path, std_msgs_py_path):
    # type: (str, str, str) -> None
    package = "std_msgs"  # type: str
    package_dir = message_path(std_msgs_path)
    package_py_dir = message_path(std_msgs_py_path)
    expected_dir = os.path.join(expected_dir, package, "msg")
 
    with temporary_directory() as td:
        cli.run_module(package_dir, td, "genmsg")
 
        assert_output_equals(expected_dir, td, "__init__.pyi")
 
    with temporary_directory() as td:
        cli.run_module(package_py_dir, td, "py")
 
        assert_output_equals(expected_dir, td, "__init__.pyi")
 
 
def test_sensor_msgs_srv(
    expected_dir, std_msgs_path, sensor_msgs_path, sensor_msgs_py_path
):
    # type: (str, str, str, str) -> None
    package = "sensor_msgs"  # type: str
    package_dir = service_path(sensor_msgs_path)
    package_py_dir = service_path(sensor_msgs_py_path)
    expected_dir = os.path.join(expected_dir, package, "srv")
 
    with temporary_directory() as td:
        cli.run_module(package_dir, td, "genmsg")
 
        assert_output_equals(expected_dir, td, "__init__.pyi")
 
    with temporary_directory() as td:
        cli.run_module(package_py_dir, td, "py")
 
        assert_output_equals(expected_dir, td, "__init__.pyi")