from compose_viz.port import Port, Protocol def test_port_init_normal() -> None: try: p = Port(host_port="8080", container_port="80") assert p.host_port == "8080" assert p.container_port == "80" assert p.protocol == Protocol.tcp except Exception as e: assert False, e def test_port_with_protocol() -> None: try: p = Port(host_port="8080", container_port="80", protocol=Protocol.udp) assert p.host_port == "8080" assert p.container_port == "80" assert p.protocol == Protocol.udp except Exception as e: assert False, e