swh:1:snp:4e3e7077647a709f15b8c1b32ce7100175d0580b
Tip revision: 0fa8ab5b81e54410ee9b34a1ac5e45eebda7e387 authored by Jean Kossaifi on 06 August 2019, 17:16:06 UTC
Upload only once
Upload only once
Tip revision: 0fa8ab5
testing.py
import numpy as np
from tensorly import backend as T
def assert_array_equal(a, b, *args, **kwargs):
np.testing.assert_array_equal(T.to_numpy(a), T.to_numpy(b),
*args, **kwargs)
def assert_array_almost_equal(a, b, *args, **kwargs):
np.testing.assert_array_almost_equal(T.to_numpy(a), T.to_numpy(b),
*args, **kwargs)
def assert_equal(actual, desired, *args, **kwargs):
def _tensor_to_numpy(x):
if T.is_tensor(x):
x = T.to_numpy(x)
return x[0] if x.shape == (1,) else x
return x
np.testing.assert_equal(_tensor_to_numpy(actual),
_tensor_to_numpy(desired),
*args, **kwargs)
assert_ = np.testing.assert_
assert_raises = np.testing.assert_raises