• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright 2022 Huawei Technologies Co., Ltd
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14# ============================================================================
15import numpy as np
16import pytest
17import mindspore as ms
18import mindspore.nn as nn
19from mindspore import Tensor
20from mindspore import ops
21
22
23class Net(nn.Cell):
24    def construct(self, x):
25        return ops.arccos(x)
26
27
28@pytest.mark.level2
29@pytest.mark.platform_x86_cpu
30@pytest.mark.platform_arm_cpu
31@pytest.mark.platform_x86_gpu_training
32@pytest.mark.platform_arm_ascend_training
33@pytest.mark.platform_x86_ascend_training
34@pytest.mark.env_onecard
35@pytest.mark.parametrize('mode', [ms.GRAPH_MODE, ms.PYNATIVE_MODE])
36def test_arccos(mode):
37    """
38    Feature: arccos
39    Description: Verify the result of arccos
40    Expectation: success
41    """
42    ms.set_context(mode=mode)
43    x = Tensor(np.array([0.74, 0.04, 0.30, 0.56]), ms.float32)
44    net = Net()
45    output = net(x)
46    expect_output = np.array([0.737726, 1.5307857, 1.2661036, 0.9764105], dtype=np.float32)
47    assert np.allclose(output.asnumpy(), expect_output, rtol=5e-03, atol=1.e-8)
48