mirror of
https://github.com/home-assistant/core.git
synced 2026-02-05 14:55:35 +01:00
Compare commits
1 Commits
edenhaus-b
...
infrared
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
785e6de8b2 |
2
CODEOWNERS
generated
2
CODEOWNERS
generated
@@ -782,6 +782,8 @@ build.json @home-assistant/supervisor
|
||||
/tests/components/inels/ @epdevlab
|
||||
/homeassistant/components/influxdb/ @mdegat01
|
||||
/tests/components/influxdb/ @mdegat01
|
||||
/homeassistant/components/infrared/ @home-assistant/core
|
||||
/tests/components/infrared/ @home-assistant/core
|
||||
/homeassistant/components/inkbird/ @bdraco
|
||||
/tests/components/inkbird/ @bdraco
|
||||
/homeassistant/components/input_boolean/ @home-assistant/core
|
||||
|
||||
157
homeassistant/components/infrared/__init__.py
Normal file
157
homeassistant/components/infrared/__init__.py
Normal file
@@ -0,0 +1,157 @@
|
||||
"""Provides functionality to interact with infrared devices."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from abc import abstractmethod
|
||||
from datetime import datetime, timedelta
|
||||
import logging
|
||||
from typing import final
|
||||
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.core import Context, HomeAssistant, callback
|
||||
from homeassistant.exceptions import HomeAssistantError
|
||||
from homeassistant.helpers import config_validation as cv, entity_registry as er
|
||||
from homeassistant.helpers.entity import EntityDescription
|
||||
from homeassistant.helpers.entity_component import EntityComponent
|
||||
from homeassistant.helpers.restore_state import RestoreEntity
|
||||
from homeassistant.helpers.typing import ConfigType
|
||||
from homeassistant.util import dt as dt_util
|
||||
from homeassistant.util.hass_dict import HassKey
|
||||
|
||||
from .const import DOMAIN
|
||||
from .protocols import InfraredCommand, InfraredProtocol, NECInfraredCommand, Timing
|
||||
|
||||
__all__ = [
|
||||
"DOMAIN",
|
||||
"InfraredCommand",
|
||||
"InfraredEntity",
|
||||
"InfraredEntityDescription",
|
||||
"InfraredProtocol",
|
||||
"NECInfraredCommand",
|
||||
"Timing",
|
||||
"async_get_entities",
|
||||
"async_send_command",
|
||||
]
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
DATA_COMPONENT: HassKey[EntityComponent[InfraredEntity]] = HassKey(DOMAIN)
|
||||
ENTITY_ID_FORMAT = DOMAIN + ".{}"
|
||||
PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA
|
||||
PLATFORM_SCHEMA_BASE = cv.PLATFORM_SCHEMA_BASE
|
||||
SCAN_INTERVAL = timedelta(seconds=30)
|
||||
|
||||
|
||||
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
|
||||
"""Set up the infrared domain."""
|
||||
component = hass.data[DATA_COMPONENT] = EntityComponent[InfraredEntity](
|
||||
_LOGGER, DOMAIN, hass, SCAN_INTERVAL
|
||||
)
|
||||
await component.async_setup(config)
|
||||
|
||||
return True
|
||||
|
||||
|
||||
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
||||
"""Set up a config entry."""
|
||||
return await hass.data[DATA_COMPONENT].async_setup_entry(entry)
|
||||
|
||||
|
||||
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
||||
"""Unload a config entry."""
|
||||
return await hass.data[DATA_COMPONENT].async_unload_entry(entry)
|
||||
|
||||
|
||||
@callback
|
||||
def async_get_emitters(hass: HomeAssistant) -> list[InfraredEntity]:
|
||||
"""Get all infrared emitters."""
|
||||
component = hass.data.get(DATA_COMPONENT)
|
||||
if component is None:
|
||||
return []
|
||||
|
||||
return list(component.entities)
|
||||
|
||||
|
||||
async def async_send_command(
|
||||
hass: HomeAssistant,
|
||||
entity_uuid: str,
|
||||
command: InfraredCommand,
|
||||
context: Context | None = None,
|
||||
) -> None:
|
||||
"""Send an IR command to the specified infrared entity.
|
||||
|
||||
Raises:
|
||||
HomeAssistantError: If the infrared entity is not found.
|
||||
"""
|
||||
component = hass.data.get(DATA_COMPONENT)
|
||||
if component is None:
|
||||
raise HomeAssistantError("Infrared component not loaded.")
|
||||
|
||||
ent_reg = er.async_get(hass)
|
||||
entity_id = er.async_validate_entity_id(ent_reg, entity_uuid)
|
||||
entity = component.get_entity(entity_id)
|
||||
if entity is None:
|
||||
raise HomeAssistantError(
|
||||
translation_domain=DOMAIN,
|
||||
translation_key="entity_not_found",
|
||||
translation_placeholders={"entity_id": entity_id},
|
||||
)
|
||||
|
||||
if context is not None:
|
||||
entity.async_set_context(context)
|
||||
|
||||
await entity.async_send_command_internal(command)
|
||||
|
||||
|
||||
class InfraredEntityDescription(EntityDescription, frozen_or_thawed=True):
|
||||
"""Describes infrared entities."""
|
||||
|
||||
|
||||
CACHED_PROPERTIES_WITH_ATTR_ = {"supported_features"}
|
||||
|
||||
|
||||
class InfraredEntity(RestoreEntity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
|
||||
"""Base class for infrared transmitter entities."""
|
||||
|
||||
entity_description: InfraredEntityDescription
|
||||
_attr_should_poll = False
|
||||
_attr_state: None
|
||||
|
||||
__last_command_sent: datetime | None = None
|
||||
|
||||
@property
|
||||
@final
|
||||
def state(self) -> str | None:
|
||||
"""Return the entity state."""
|
||||
if (last_command := self.__last_command_sent) is None:
|
||||
return None
|
||||
return last_command.isoformat(timespec="milliseconds")
|
||||
|
||||
@final
|
||||
async def async_send_command_internal(self, command: InfraredCommand) -> None:
|
||||
"""Send an IR command and update state.
|
||||
|
||||
Should not be overridden, handles setting last sent timestamp.
|
||||
"""
|
||||
self.__last_command_sent = dt_util.utcnow()
|
||||
self.async_write_ha_state()
|
||||
await self.async_send_command(command)
|
||||
|
||||
@final
|
||||
async def async_internal_added_to_hass(self) -> None:
|
||||
"""Call when the infrared entity is added to hass."""
|
||||
await super().async_internal_added_to_hass()
|
||||
state = await self.async_get_last_state()
|
||||
if state is not None and state.state is not None:
|
||||
self.__last_command_sent = dt_util.parse_datetime(state.state)
|
||||
|
||||
@abstractmethod
|
||||
async def async_send_command(self, command: InfraredCommand) -> None:
|
||||
"""Send an IR command.
|
||||
|
||||
Args:
|
||||
command: The IR command to send.
|
||||
|
||||
Raises:
|
||||
HomeAssistantError: If transmission fails.
|
||||
"""
|
||||
5
homeassistant/components/infrared/const.py
Normal file
5
homeassistant/components/infrared/const.py
Normal file
@@ -0,0 +1,5 @@
|
||||
"""Constants for the Infrared integration."""
|
||||
|
||||
from typing import Final
|
||||
|
||||
DOMAIN: Final = "infrared"
|
||||
8
homeassistant/components/infrared/manifest.json
Normal file
8
homeassistant/components/infrared/manifest.json
Normal file
@@ -0,0 +1,8 @@
|
||||
{
|
||||
"domain": "infrared",
|
||||
"name": "Infrared",
|
||||
"codeowners": ["@home-assistant/core"],
|
||||
"documentation": "https://www.home-assistant.io/integrations/infrared",
|
||||
"integration_type": "entity",
|
||||
"quality_scale": "internal"
|
||||
}
|
||||
124
homeassistant/components/infrared/protocols.py
Normal file
124
homeassistant/components/infrared/protocols.py
Normal file
@@ -0,0 +1,124 @@
|
||||
"""IR protocol definitions for the Infrared integration."""
|
||||
|
||||
import abc
|
||||
from dataclasses import dataclass
|
||||
from enum import StrEnum
|
||||
from typing import override
|
||||
|
||||
|
||||
class InfraredProtocol(StrEnum):
|
||||
"""IR protocol type identifiers."""
|
||||
|
||||
NEC = "nec"
|
||||
|
||||
|
||||
@dataclass(frozen=True, slots=True)
|
||||
class Timing:
|
||||
"""High/low signal timing."""
|
||||
|
||||
high_us: int
|
||||
low_us: int
|
||||
|
||||
|
||||
class InfraredCommand(abc.ABC):
|
||||
"""Base class for IR commands."""
|
||||
|
||||
protocol: InfraredProtocol
|
||||
repeat_count: int
|
||||
modulation: int
|
||||
|
||||
def __init__(self, *, modulation: int, repeat_count: int = 0) -> None:
|
||||
"""Initialize the IR command."""
|
||||
self.modulation = modulation
|
||||
self.repeat_count = repeat_count
|
||||
|
||||
@abc.abstractmethod
|
||||
def get_raw_timings(self) -> list[Timing]:
|
||||
"""Get raw timings for the command."""
|
||||
|
||||
|
||||
class NECInfraredCommand(InfraredCommand):
|
||||
"""NEC IR command."""
|
||||
|
||||
protocol = InfraredProtocol.NEC
|
||||
address: int
|
||||
command: int
|
||||
|
||||
def __init__(
|
||||
self, *, address: int, command: int, modulation: int, repeat_count: int = 0
|
||||
) -> None:
|
||||
"""Initialize the NEC IR command."""
|
||||
super().__init__(modulation=modulation, repeat_count=repeat_count)
|
||||
self.address = address
|
||||
self.command = command
|
||||
|
||||
@override
|
||||
def get_raw_timings(self) -> list[Timing]:
|
||||
"""Get raw timings for the NEC command.
|
||||
|
||||
NEC protocol timing (in microseconds):
|
||||
- Leader pulse: 9000µs high, 4500µs low
|
||||
- Logical '0': 562µs high, 562µs low
|
||||
- Logical '1': 562µs high, 1687µs low
|
||||
- End pulse: 562µs high
|
||||
- Repeat code: 9000µs high, 2250µs low, 562µs end pulse
|
||||
- Frame gap: ~96ms between end pulse and next frame (total frame ~108ms)
|
||||
|
||||
Data format (32 bits, LSB first):
|
||||
- Standard NEC: address (8-bit) + ~address (8-bit) + command (8-bit) + ~command (8-bit)
|
||||
- Extended NEC: address_low (8-bit) + address_high (8-bit) + command (8-bit) + ~command (8-bit)
|
||||
"""
|
||||
# NEC timing constants (microseconds)
|
||||
leader_high = 9000
|
||||
leader_low = 4500
|
||||
bit_high = 562
|
||||
zero_low = 562
|
||||
one_low = 1687
|
||||
repeat_low = 2250
|
||||
frame_gap = 96000 # Gap to make total frame ~108ms
|
||||
|
||||
timings: list[Timing] = [Timing(high_us=leader_high, low_us=leader_low)]
|
||||
|
||||
# Determine if standard (8-bit) or extended (16-bit) address
|
||||
if self.address <= 0xFF:
|
||||
# Standard NEC: address + inverted address
|
||||
address_low = self.address & 0xFF
|
||||
address_high = (~self.address) & 0xFF
|
||||
else:
|
||||
# Extended NEC: 16-bit address (no inversion)
|
||||
address_low = self.address & 0xFF
|
||||
address_high = (self.address >> 8) & 0xFF
|
||||
|
||||
command_byte = self.command & 0xFF
|
||||
command_inverted = (~self.command) & 0xFF
|
||||
|
||||
# Build 32-bit command data (LSB first in transmission)
|
||||
data = (
|
||||
address_low
|
||||
| (address_high << 8)
|
||||
| (command_byte << 16)
|
||||
| (command_inverted << 24)
|
||||
)
|
||||
|
||||
for _ in range(32):
|
||||
bit = data & 1
|
||||
if bit:
|
||||
timings.append(Timing(high_us=bit_high, low_us=one_low))
|
||||
else:
|
||||
timings.append(Timing(high_us=bit_high, low_us=zero_low))
|
||||
data >>= 1
|
||||
|
||||
# End pulse
|
||||
timings.append(Timing(high_us=bit_high, low_us=0))
|
||||
|
||||
# Add repeat codes if requested
|
||||
for _ in range(self.repeat_count):
|
||||
# Replace the last timing's low_us with the frame gap
|
||||
last_timing = timings[-1]
|
||||
timings[-1] = Timing(high_us=last_timing.high_us, low_us=frame_gap)
|
||||
|
||||
# Repeat code: leader burst + shorter space + end pulse
|
||||
timings.append(Timing(high_us=leader_high, low_us=repeat_low))
|
||||
timings.append(Timing(high_us=bit_high, low_us=0))
|
||||
|
||||
return timings
|
||||
10
homeassistant/components/infrared/strings.json
Normal file
10
homeassistant/components/infrared/strings.json
Normal file
@@ -0,0 +1,10 @@
|
||||
{
|
||||
"exceptions": {
|
||||
"entity_not_found": {
|
||||
"message": "Infrared entity `{entity_id}` not found"
|
||||
},
|
||||
"send_command_failed": {
|
||||
"message": "Failed to send IR command: {error}"
|
||||
}
|
||||
}
|
||||
}
|
||||
1
homeassistant/generated/entity_platforms.py
generated
1
homeassistant/generated/entity_platforms.py
generated
@@ -29,6 +29,7 @@ class EntityPlatforms(StrEnum):
|
||||
HUMIDIFIER = "humidifier"
|
||||
IMAGE = "image"
|
||||
IMAGE_PROCESSING = "image_processing"
|
||||
INFRARED = "infrared"
|
||||
LAWN_MOWER = "lawn_mower"
|
||||
LIGHT = "light"
|
||||
LOCK = "lock"
|
||||
|
||||
1
tests/components/infrared/__init__.py
Normal file
1
tests/components/infrared/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""Tests for the Infrared integration."""
|
||||
37
tests/components/infrared/conftest.py
Normal file
37
tests/components/infrared/conftest.py
Normal file
@@ -0,0 +1,37 @@
|
||||
"""Common fixtures for the Infrared tests."""
|
||||
|
||||
import pytest
|
||||
|
||||
from homeassistant.components.infrared import InfraredCommand, InfraredEntity
|
||||
from homeassistant.components.infrared.const import DOMAIN
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
async def init_integration(hass: HomeAssistant) -> None:
|
||||
"""Set up the Infrared integration for testing."""
|
||||
assert await async_setup_component(hass, DOMAIN, {})
|
||||
await hass.async_block_till_done()
|
||||
|
||||
|
||||
class MockInfraredEntity(InfraredEntity):
|
||||
"""Mock infrared entity for testing."""
|
||||
|
||||
_attr_has_entity_name = True
|
||||
_attr_name = "Test IR transmitter"
|
||||
|
||||
def __init__(self, unique_id: str) -> None:
|
||||
"""Initialize mock entity."""
|
||||
self._attr_unique_id = unique_id
|
||||
self.send_command_calls: list[InfraredCommand] = []
|
||||
|
||||
async def async_send_command(self, command: InfraredCommand) -> None:
|
||||
"""Mock send command."""
|
||||
self.send_command_calls.append(command)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_infrared_entity() -> MockInfraredEntity:
|
||||
"""Return a mock infrared entity."""
|
||||
return MockInfraredEntity("test_ir_transmitter")
|
||||
118
tests/components/infrared/test_init.py
Normal file
118
tests/components/infrared/test_init.py
Normal file
@@ -0,0 +1,118 @@
|
||||
"""Tests for the Infrared integration setup."""
|
||||
|
||||
from freezegun.api import FrozenDateTimeFactory
|
||||
import pytest
|
||||
|
||||
from homeassistant.components.infrared import (
|
||||
DATA_COMPONENT,
|
||||
DOMAIN,
|
||||
NECInfraredCommand,
|
||||
async_get_emitters,
|
||||
async_send_command,
|
||||
)
|
||||
from homeassistant.const import STATE_UNKNOWN
|
||||
from homeassistant.core import HomeAssistant, State
|
||||
from homeassistant.exceptions import HomeAssistantError
|
||||
from homeassistant.setup import async_setup_component
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from .conftest import MockInfraredEntity
|
||||
|
||||
from tests.common import mock_restore_cache
|
||||
|
||||
|
||||
async def test_get_entities_integration_setup(hass: HomeAssistant) -> None:
|
||||
"""Test getting entities when the integration is not setup."""
|
||||
assert async_get_emitters(hass) == []
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("init_integration")
|
||||
async def test_get_entities_empty(hass: HomeAssistant) -> None:
|
||||
"""Test getting entities when none are registered."""
|
||||
assert async_get_emitters(hass) == []
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("init_integration")
|
||||
async def test_infrared_entity_initial_state(
|
||||
hass: HomeAssistant, mock_infrared_entity: MockInfraredEntity
|
||||
) -> None:
|
||||
"""Test infrared entity has no state before any command is sent."""
|
||||
component = hass.data[DATA_COMPONENT]
|
||||
await component.async_add_entities([mock_infrared_entity])
|
||||
|
||||
state = hass.states.get("infrared.test_ir_transmitter")
|
||||
assert state is not None
|
||||
assert state.state == STATE_UNKNOWN
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("init_integration")
|
||||
async def test_async_send_command_success(
|
||||
hass: HomeAssistant,
|
||||
mock_infrared_entity: MockInfraredEntity,
|
||||
freezer: FrozenDateTimeFactory,
|
||||
) -> None:
|
||||
"""Test sending command via async_send_command helper."""
|
||||
# Add the mock entity to the component
|
||||
component = hass.data[DATA_COMPONENT]
|
||||
await component.async_add_entities([mock_infrared_entity])
|
||||
|
||||
# Freeze time so we can verify the state update
|
||||
now = dt_util.utcnow()
|
||||
freezer.move_to(now)
|
||||
|
||||
command = NECInfraredCommand(
|
||||
address=0x04FB, command=0x08F7, modulation=38000, repeat_count=1
|
||||
)
|
||||
await async_send_command(hass, mock_infrared_entity.entity_id, command)
|
||||
|
||||
assert len(mock_infrared_entity.send_command_calls) == 1
|
||||
assert mock_infrared_entity.send_command_calls[0] is command
|
||||
|
||||
state = hass.states.get("infrared.test_ir_transmitter")
|
||||
assert state is not None
|
||||
assert state.state == now.isoformat(timespec="milliseconds")
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("init_integration")
|
||||
async def test_async_send_command_entity_not_found(hass: HomeAssistant) -> None:
|
||||
"""Test async_send_command raises error when entity not found."""
|
||||
command = NECInfraredCommand(
|
||||
address=0x04FB, command=0x08F7, modulation=38000, repeat_count=1
|
||||
)
|
||||
|
||||
with pytest.raises(
|
||||
HomeAssistantError,
|
||||
match="Infrared entity `infrared.nonexistent_entity` not found",
|
||||
):
|
||||
await async_send_command(hass, "infrared.nonexistent_entity", command)
|
||||
|
||||
|
||||
async def test_async_send_command_component_not_loaded(hass: HomeAssistant) -> None:
|
||||
"""Test async_send_command raises error when component not loaded."""
|
||||
command = NECInfraredCommand(
|
||||
address=0x04FB, command=0x08F7, modulation=38000, repeat_count=1
|
||||
)
|
||||
|
||||
with pytest.raises(HomeAssistantError, match="Infrared component not loaded"):
|
||||
await async_send_command(hass, "infrared.some_entity", command)
|
||||
|
||||
|
||||
async def test_infrared_entity_state_restore(
|
||||
hass: HomeAssistant,
|
||||
mock_infrared_entity: MockInfraredEntity,
|
||||
) -> None:
|
||||
"""Test infrared entity restores state from previous session."""
|
||||
previous_timestamp = "2026-01-01T12:00:00.000+00:00"
|
||||
mock_restore_cache(
|
||||
hass, [State("infrared.test_ir_transmitter", previous_timestamp)]
|
||||
)
|
||||
|
||||
assert await async_setup_component(hass, DOMAIN, {})
|
||||
await hass.async_block_till_done()
|
||||
|
||||
component = hass.data[DATA_COMPONENT]
|
||||
await component.async_add_entities([mock_infrared_entity])
|
||||
|
||||
state = hass.states.get("infrared.test_ir_transmitter")
|
||||
assert state is not None
|
||||
assert state.state == previous_timestamp
|
||||
138
tests/components/infrared/test_protocols.py
Normal file
138
tests/components/infrared/test_protocols.py
Normal file
@@ -0,0 +1,138 @@
|
||||
"""Tests for the Infrared protocol definitions."""
|
||||
|
||||
from homeassistant.components.infrared import (
|
||||
InfraredProtocol,
|
||||
NECInfraredCommand,
|
||||
Timing,
|
||||
)
|
||||
|
||||
|
||||
def test_nec_command_get_raw_timings_standard() -> None:
|
||||
"""Test NEC command raw timings generation for standard 8-bit address."""
|
||||
expected_raw_timings = [
|
||||
Timing(high_us=9000, low_us=4500),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=0),
|
||||
]
|
||||
command = NECInfraredCommand(
|
||||
address=0x04, command=0x08, modulation=38000, repeat_count=0
|
||||
)
|
||||
timings = command.get_raw_timings()
|
||||
assert timings == expected_raw_timings
|
||||
|
||||
# Same command now with 2 repeats
|
||||
command_with_repeats = NECInfraredCommand(
|
||||
address=command.address,
|
||||
command=command.command,
|
||||
modulation=command.modulation,
|
||||
repeat_count=2,
|
||||
)
|
||||
timings_with_repeats = command_with_repeats.get_raw_timings()
|
||||
assert timings_with_repeats == [
|
||||
*expected_raw_timings[:-1],
|
||||
Timing(high_us=562, low_us=96000),
|
||||
Timing(high_us=9000, low_us=2250),
|
||||
Timing(high_us=562, low_us=96000),
|
||||
Timing(high_us=9000, low_us=2250),
|
||||
Timing(high_us=562, low_us=0),
|
||||
]
|
||||
|
||||
|
||||
def test_nec_command_get_raw_timings_extended() -> None:
|
||||
"""Test NEC command raw timings generation for extended 16-bit address."""
|
||||
expected_raw_timings = [
|
||||
Timing(high_us=9000, low_us=4500),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=562),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=1687),
|
||||
Timing(high_us=562, low_us=0),
|
||||
]
|
||||
|
||||
command = NECInfraredCommand(
|
||||
address=0x04FB, command=0x08, modulation=38000, repeat_count=0
|
||||
)
|
||||
timings = command.get_raw_timings()
|
||||
assert timings == expected_raw_timings
|
||||
|
||||
# Same command now with 2 repeats
|
||||
command_with_repeats = NECInfraredCommand(
|
||||
address=command.address,
|
||||
command=command.command,
|
||||
modulation=command.modulation,
|
||||
repeat_count=2,
|
||||
)
|
||||
timings_with_repeats = command_with_repeats.get_raw_timings()
|
||||
assert timings_with_repeats == [
|
||||
*expected_raw_timings[:-1],
|
||||
Timing(high_us=562, low_us=96000),
|
||||
Timing(high_us=9000, low_us=2250),
|
||||
Timing(high_us=562, low_us=96000),
|
||||
Timing(high_us=9000, low_us=2250),
|
||||
Timing(high_us=562, low_us=0),
|
||||
]
|
||||
|
||||
|
||||
def test_nec_command_protocol_attribute() -> None:
|
||||
"""Test NEC command has correct protocol attribute."""
|
||||
command = NECInfraredCommand(address=0x04, command=0x08, modulation=38000)
|
||||
assert command.protocol == InfraredProtocol.NEC
|
||||
Reference in New Issue
Block a user