def test_round_away(): float_dtype = np.float32 data = ng.parameter(Shape([3, 10]), dtype=float_dtype, name="data") node = ng.round(data, "HALF_AWAY_FROM_ZERO") assert node.get_type_name() == "Round" assert node.get_output_size() == 1 assert list(node.get_output_shape(0)) == [3, 10] assert node.get_output_element_type(0) == Type.f32
def test_round_away(): float_dtype = np.float32 data = ng.parameter(Shape([3, 10]), dtype=float_dtype, name="data") node = ng.round(data, "HALF_AWAY_FROM_ZERO") assert node.get_type_name() == "Round" assert node.get_output_size() == 1 assert list(node.get_output_shape(0)) == [3, 10] assert node.get_output_element_type(0) == Type.f32 input_tensor = np.array([-2.5, -1.5, -0.5, 0.5, 0.9, 1.5, 2.3, 2.5, 3.5], dtype=np.float32) expected = [-3.0, -2.0, -1.0, 1.0, 1.0, 2.0, 2.0, 3.0, 4.0] result = run_op_node([input_tensor], ng.round, "HALF_AWAY_FROM_ZERO") assert np.allclose(result, expected)