tests: audio: Add unit test for tensorflow

Add ZTest unit test validating module creation, configuration,
prepare, copy, and teardown lifecycle for the tensorflow component.

Signed-off-by: Liam Girdwood <liam.r.girdwood@linux.intel.com>

Author: lrgirdwo

zephyr/test/audio/test_tensorflow.c

@@ -0,0 +1,138 @@
+// SPDX-License-Identifier: BSD-3-Clause
+/*
+ * Copyright(c) 2026 Intel Corporation.
+ */
+
+#include <zephyr/kernel.h>
+#include <zephyr/ztest.h>
+#include <string.h>
+#include <rtos/sof.h>
+#include <sof/list.h>
+#include <sof/audio/component.h>
+#include <sof/audio/component_ext.h>
+#include <sof/ipc/topology.h>
+#include <ipc4/base-config.h>
+
+#include <rtos/alloc.h>
+#include "test_audio_helper.h"
+
+SOF_DEFINE_REG_UUID(tflmcly);
+extern const struct sof_uuid tflmcly_uuid;
+
+extern void sys_comp_module_tflmcly_interface_init(void);
+
+static void *setup(void)
+{
+	struct sof *sof = sof_get();
+
+	sys_comp_init(sof);
+
+	if (!sof->ipc) {
+		sof->ipc = rzalloc(SOF_MEM_FLAG_COHERENT, sizeof(*sof->ipc));
+		sof->ipc->comp_data = rzalloc(SOF_MEM_FLAG_COHERENT, 4096);
+		k_spinlock_init(&sof->ipc->lock);
+		list_init(&sof->ipc->msg_list);
+		list_init(&sof->ipc->comp_list);
+	}
+
+#if defined(CONFIG_COMP_TENSORFLOW)
+	sys_comp_module_tflmcly_interface_init();
+#endif
+	return NULL;
+}
+
+extern const struct sof_uuid tflmcly_uuid;
+
+struct custom_ipc4_config_tensorflow {
+	struct ipc4_base_module_cfg base;
+} __attribute__((packed, aligned(8)));
+
+static struct comp_dev *test_tensorflow_create(void)
+{
+#ifndef CONFIG_COMP_TENSORFLOW
+	ztest_test_skip();
+#endif
+	struct comp_dev *comp;
+	struct comp_ipc_config ipc_config;
+	struct ipc_config_process spec;
+	struct custom_ipc4_config_tensorflow init_data;
+
+	memset(&init_data, 0, sizeof(init_data));
+	init_data.base.audio_fmt.channels_count = 1;
+	init_data.base.audio_fmt.sampling_frequency = 16000;
+	init_data.base.audio_fmt.depth = 16;
+	init_data.base.audio_fmt.valid_bit_depth = 16;
+	init_data.base.audio_fmt.interleaving_style = IPC4_CHANNELS_INTERLEAVED;
+	init_data.base.ibs = 1024;
+	init_data.base.obs = 1024;
+
+	memset(&ipc_config, 0, sizeof(ipc_config));
+	ipc_config.id = 1;
+	ipc_config.pipeline_id = 1;
+	ipc_config.core = 0;
+
+	memset(&spec, 0, sizeof(spec));
+	spec.size = sizeof(init_data);
+	spec.data = (unsigned char *)&init_data;
+
+	struct list_item *clist;
+	const struct comp_driver *drv = NULL;
+
+	list_for_item(clist, &comp_drivers_get()->list) {
+		struct comp_driver_info *info = container_of(clist, struct comp_driver_info, list);
+		if (!info->drv->uid) continue;
+		if (!memcmp(info->drv->uid, &tflmcly_uuid, sizeof(struct sof_uuid))) {
+			drv = info->drv;
+			break;
+		}
+	}
+	zassert_not_null(drv, "Driver for tensorflow missing");
+
+	comp = drv->ops.create(drv, &ipc_config, &spec);
+	zassert_not_null(comp, "Component allocation failed");
+
+	return comp;
+}
+
+/* Test tensorflow initialization */
+ZTEST(audio_tensorflow, test_tensorflow_init)
+{
+	struct comp_dev *comp = test_tensorflow_create();
+	zassert_equal(comp->state, COMP_STATE_READY, "Component is not in READY state");
+
+		comp->drv->ops.free(comp);
+}
+ZTEST(audio_tensorflow, test_tensorflow_config)
+{
+	struct comp_dev *comp = test_tensorflow_create();
+
+	int ret = 0;
+	if (comp->drv->ops.set_large_config) {
+		uint32_t val = 0;
+		ret = comp->drv->ops.set_large_config(comp, 0, true, true,
+			sizeof(val), (uint8_t *)&val);
+	}
+
+	comp->drv->ops.free(comp);
+}
+
+ZTEST(audio_tensorflow, test_tensorflow_process)
+{
+	struct comp_dev *comp = test_tensorflow_create();
+
+	struct sof_ipc_stream_params params = {0};
+	params.buffer_fmt = SOF_IPC_BUFFER_INTERLEAVED;
+	params.channels = 2;
+	params.rate = 48000;
+	params.sample_container_bytes = 4;
+	params.sample_valid_bytes = 4;
+
+	test_audio_helper_setup_buffers(comp, 4096, &params);
+	test_audio_helper_process(comp);
+	test_audio_helper_free_buffers(comp);
+
+	comp->drv->ops.free(comp);
+}
+
+
+ZTEST_SUITE(audio_tensorflow, NULL, setup, NULL, NULL, NULL);