From a0afc8f626e93a4d5fb4886d44108dffa8f13c45 Mon Sep 17 00:00:00 2001
From: SushRaja <me.sushanth@gmail.com>
Date: Sun, 9 Jul 2023 11:42:36 -0700
Subject: [PATCH] Add a test'

---
 tests/node_converters/depth_to_space_test.py | 28 ++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 tests/node_converters/depth_to_space_test.py

diff --git a/tests/node_converters/depth_to_space_test.py b/tests/node_converters/depth_to_space_test.py
new file mode 100644
index 00000000..f95c017d
--- /dev/null
+++ b/tests/node_converters/depth_to_space_test.py
@@ -0,0 +1,28 @@
+from typing import List
+
+import numpy as np
+import onnx
+
+from tests.utils.common import check_onnx_model
+from tests.utils.common import make_model_from_nodes
+
+
+def _test_depth_to_space(
+    input_shape: List[int],
+    **kwargs,
+) -> None:
+    x = np.random.uniform(low=-1.0, high=1.0, size=input_shape).astype(np.float32)
+    test_inputs = {'x': x}
+
+    node = onnx.helper.make_node(
+        op_type='DepthToSpace',
+        inputs=['x'],
+        outputs=['y'],
+        **kwargs,
+    )
+    model = make_model_from_nodes(nodes=node, initializers={}, inputs_example=test_inputs, opset_version=13)
+    check_onnx_model(model, test_inputs)
+
+
+def test_depth_to_space() -> None:  # pylint: disable=missing-function-docstring
+    _test_depth_to_space(input_shape=[1, 12, 3, 3], blocksize=2, mode='CRD')
\ No newline at end of file