MegEngine/python_module/test/unit/functional/test_elemwise.py

# -*- coding: utf-8 -*-
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) 2014-2020 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import numpy as np

import megengine.functional as F
from megengine import tensor

from megengine.test import assertTensorClose


def test_abs():
    assertTensorClose(
        F.abs(tensor([-3., -4., -5.])).numpy(),
        np.abs(np.array([-3., -4., -5.], dtype=np.float32)))

    assertTensorClose(F.abs(-3.), np.abs(np.float32(-3.)))


def test_multiply():
    assertTensorClose(F.multiply(-3., -4.),
                      np.multiply(np.float32(-3.), np.float32(-4.)))

    assertTensorClose(
        F.multiply(tensor([3., 4.]), 4.).numpy(),
        np.multiply(np.array([3., 4.], dtype=np.float32), 4.))

    assertTensorClose(
        F.multiply(4., tensor([3., 4.])).numpy(),
        np.multiply(4., np.array([3., 4.], dtype=np.float32)))

    assertTensorClose(
        F.multiply(tensor([3., 4.]), tensor([3., 4.])).numpy(),
        np.multiply(np.array([3., 4.], dtype=np.float32),
                    np.array([3., 4.], dtype=np.float32)))