tests: benchmarks: pm_states: Add pm_states test

Add a test to check if power management states are switched correctly.

Signed-off-by: Piotr Krzyzanowski <[email protected]>

Author: nordic-pikr

scripts/ci/tags.yaml

@@ -1244,6 +1244,7 @@ ci_tests_benchmarks_current_consumption:
     - nrf/include/nfc/
     - nrf/subsys/nfc/
     - nrf/tests/benchmarks/current_consumption/
+    - nrf/tests/benchmarks/pm_states/
     - nrf/tests/benchmarks/power_consumption/
     - nrfxlib/nfc/
     - zephyr/boards/nordic/

tests/benchmarks/pm_states/CMakeLists.txt

@@ -0,0 +1,19 @@
+#
+# Copyright (c) 2025 Nordic Semiconductor ASA
+#
+# SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
+#
+
+cmake_minimum_required(VERSION 3.20.0)
+
+find_package(Zephyr REQUIRED HINTS $ENV{ZEPHYR_BASE})
+
+if(NOT SYSBUILD)
+  message(FATAL_ERROR
+    " This is a multi-image application that should be built using sysbuild.\n"
+    " Add --sysbuild argument to west build command to prepare all the images.")
+endif()
+
+project(pm_states)
+
+target_sources(app PRIVATE src/main.c)

tests/benchmarks/pm_states/Kconfig.sysbuild

@@ -0,0 +1,10 @@
+#
+# Copyright (c) 2025 Nordic Semiconductor ASA
+#
+# SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
+#
+
+config NRF_DEFAULT_EMPTY
+	default y
+
+source "$(ZEPHYR_BASE)/share/sysbuild/Kconfig"

tests/benchmarks/pm_states/boards/nrf54h20dk_nrf54h20_cpuapp.conf

@@ -0,0 +1,7 @@
+#
+# Copyright (c) 2025 Nordic Semiconductor ASA
+#
+# SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
+#
+
+CONFIG_SOC_NRF54H20_CPURAD_ENABLE=y

tests/benchmarks/pm_states/boards/nrf54h20dk_nrf54h20_cpuapp.overlay

@@ -0,0 +1,9 @@
+/*
+ * Copyright (c) 2025 Nordic Semiconductor ASA
+ *
+ * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
+ */
+
+&uart136 {
+	status = "disabled";
+};

tests/benchmarks/pm_states/prj.conf

@@ -0,0 +1,6 @@
+CONFIG_ASSERT=y
+CONFIG_PM=y
+CONFIG_GPIO=y
+CONFIG_CONSOLE=n
+CONFIG_UART_CONSOLE=n
+CONFIG_SERIAL=n

tests/benchmarks/pm_states/src/main.c

@@ -0,0 +1,86 @@
+/*
+ * Copyright (c) 2025 Nordic Semiconductor ASA
+ *
+ * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
+ */
+
+#include <zephyr/devicetree.h>
+#include <zephyr/kernel.h>
+#include <zephyr/drivers/gpio.h>
+#include <zephyr/pm/pm.h>
+
+#define PS_NODE DT_PATH(cpus, power_states)
+
+#if DT_NODE_EXISTS(PS_NODE)
+#define PS_VAL(node) DT_PROP_OR(node, min_residency_us, 0),
+
+static const uint32_t min_residencies[] = {DT_FOREACH_CHILD(PS_NODE, PS_VAL)};
+
+#define PS_COUNT ARRAY_SIZE(min_residencies)
+#else
+static const uint32_t min_residencies[] = {};
+#define PS_COUNT 0
+#endif
+
+static const struct gpio_dt_spec leds[] = {
+	GPIO_DT_SPEC_GET(DT_ALIAS(led0), gpios),
+	GPIO_DT_SPEC_GET(DT_ALIAS(led1), gpios),
+	GPIO_DT_SPEC_GET(DT_ALIAS(led2), gpios),
+	GPIO_DT_SPEC_GET(DT_ALIAS(led3), gpios),
+};
+#define LED_COUNT ARRAY_SIZE(leds)
+
+static void notify_pm_state_entry(enum pm_state state)
+{
+	switch (state) {
+	case PM_STATE_SUSPEND_TO_IDLE:
+		gpio_pin_set_dt(&leds[0], 1);
+		break;
+	case PM_STATE_SUSPEND_TO_RAM:
+		gpio_pin_set_dt(&leds[2], 1);
+		break;
+	default:
+		__ASSERT(true, "Unexpected PM state: %d", state);
+		break;
+	}
+}
+
+static void notify_pm_state_exit(enum pm_state state)
+{
+	for (size_t i = 0; i < LED_COUNT; i++) {
+		gpio_pin_set_dt(&leds[i], 0);
+	}
+}
+
+static struct pm_notifier notifier = {
+	.state_entry = notify_pm_state_entry,
+	.state_exit = notify_pm_state_exit,
+};
+
+int main(void)
+{
+	int ret;
+
+	for (size_t i = 0; i < LED_COUNT; i++) {
+		ret = gpio_is_ready_dt(&leds[i]);
+		__ASSERT(ret, "Error: GPIO Device not ready");
+		ret = gpio_pin_configure_dt(&leds[i], GPIO_OUTPUT_INACTIVE);
+		__ASSERT(ret == 0, "Could not configure led GPIO");
+	}
+
+	pm_notifier_register(&notifier);
+
+	k_msleep(1500);
+
+	if (PS_COUNT != 0) {
+		while (1) {
+			k_usleep(min_residencies[0] - 100);
+			for (size_t i = 0; i < PS_COUNT; i++) {
+				k_usleep(min_residencies[i] + 100);
+			}
+			k_busy_wait(400);
+		}
+	}
+
+	return 0;
+}

tests/benchmarks/pm_states/testcase.yaml

@@ -0,0 +1,14 @@
+tests:
+  benchmarks.pm_states:
+    sysbuild: true
+    platform_allow:
+      - nrf54h20dk/nrf54h20/cpuapp
+    integration_platforms:
+      - nrf54h20dk/nrf54h20/cpuapp
+    harness: pytest
+    harness_config:
+      pytest_root:
+        - "${CUSTOM_ROOT_TEST_DIR}/test_measure_power_consumption.py::test_measure_and_data_dump_pm_states"
+    tags:
+      - ci_tests_benchmarks_current_consumption
+      - ppk_power_measure