def test_euclidean_distances_op():
x = mt.random.rand(10, 3)
xx = mt.random.rand(1, 10)
y = mt.random.rand(11, 3)
d = euclidean_distances(x, X_norm_squared=xx)
assert d.op.x_norm_squared.key == check_array(xx).T.key
d = euclidean_distances(x,
y,
X_norm_squared=mt.random.rand(10,
1,
dtype=mt.float32),
Y_norm_squared=mt.random.rand(1,
11,
dtype=mt.float32))
assert d.op.x_norm_squared is None
assert d.op.y_norm_squared is None
# XX shape incompatible
with pytest.raises(ValueError):
euclidean_distances(x, X_norm_squared=mt.random.rand(10))
# XX shape incompatible
with pytest.raises(ValueError):
euclidean_distances(x, X_norm_squared=mt.random.rand(11, 1))
# YY shape incompatible
with pytest.raises(ValueError):
euclidean_distances(x, y, Y_norm_squared=mt.random.rand(10))
def testEuclideanDistancesOp(self):
x = mt.random.rand(10, 3)
xx = mt.random.rand(1, 10)
y = mt.random.rand(11, 3)
d = euclidean_distances(x, X_norm_squared=xx)
self.assertEqual(d.op.x_norm_squared.key, check_array(xx).T.key)
d = euclidean_distances(
x,
y,
X_norm_squared=mt.random.rand(10, 1, dtype=mt.float32),
Y_norm_squared=mt.random.rand(1, 11, dtype=mt.float32))
self.assertIsNone(d.op.x_norm_squared)
self.assertIsNone(d.op.y_norm_squared)
# XX shape incompatible
with self.assertRaises(ValueError):
euclidean_distances(x, X_norm_squared=mt.random.rand(10))
# XX shape incompatible
with self.assertRaises(ValueError):
euclidean_distances(x, X_norm_squared=mt.random.rand(11, 1))
# YY shape incompatible
with self.assertRaises(ValueError):
euclidean_distances(x, y, Y_norm_squared=mt.random.rand(10))