From c8d46b6686caa186c79496c624577a531d5a99c2 Mon Sep 17 00:00:00 2001 From: Marisa Kirisame Date: Mon, 15 Apr 2019 15:13:27 -0700 Subject: [PATCH] save --- tests/python/relay/test_backend_interpreter.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/python/relay/test_backend_interpreter.py b/tests/python/relay/test_backend_interpreter.py index da794e25ab56..bf7d96194bf2 100644 --- a/tests/python/relay/test_backend_interpreter.py +++ b/tests/python/relay/test_backend_interpreter.py @@ -50,6 +50,12 @@ def test_tuple_value(): np.testing.assert_allclose(tv[2].asnumpy(), 3) +def test_tuple_getitem(): + two = relay.add(relay.const(1), relay.const(1)) + func = relay.Function([], relay.TupleGetItem(relay.Tuple([relay.const(1), relay.const(2)]), 0)) + check_eval(func, [], 1) + + def test_id(): x = relay.var('x', 'float32') ident = relay.Function([x], x) @@ -181,3 +187,5 @@ def test_kwargs_params(): test_kwargs_params() test_ref() test_tensor_value() + test_tuple_value() + test_tuple_getitem()