onnx-source-snapshot/onnx/backend/test/case/node/string_concat.py

# Copyright (c) ONNX Project Contributors
#
# SPDX-License-Identifier: Apache-2.0
from __future__ import annotations

import numpy as np

import onnx
from onnx.backend.test.case.base import Base
from onnx.backend.test.case.node import expect


class StringConcat(Base):
    @staticmethod
    def export() -> None:
        node = onnx.helper.make_node(
            "StringConcat",
            inputs=["x", "y"],
            outputs=["result"],
        )
        x = np.array(["abc", "def"]).astype("object")
        y = np.array([".com", ".net"]).astype("object")
        result = np.array(["abc.com", "def.net"]).astype("object")

        expect(node, inputs=[x, y], outputs=[result], name="test_string_concat")

        x = np.array(["cat", "dog", "snake"]).astype("object")
        y = np.array(["s"]).astype("object")
        result = np.array(["cats", "dogs", "snakes"]).astype("object")

        expect(
            node,
            inputs=[x, y],
            outputs=[result],
            name="test_string_concat_broadcasting",
        )

        x = np.array("cat").astype("object")
        y = np.array("s").astype("object")
        result = np.array("cats").astype("object")

        expect(
            node,
            inputs=[x, y],
            outputs=[result],
            name="test_string_concat_zero_dimensional",
        )

        x = np.array(["abc", ""]).astype("object")
        y = np.array(["", "abc"]).astype("object")
        result = np.array(["abc", "abc"]).astype("object")

        expect(
            node,
            inputs=[x, y],
            outputs=[result],
            name="test_string_concat_empty_string",
        )

        x = np.array(["的", "中"]).astype("object")
        y = np.array(["的", "中"]).astype("object")
        result = np.array(["的的", "中中"]).astype("object")

        expect(
            node,
            inputs=[x, y],
            outputs=[result],
            name="test_string_concat_utf8",
        )