def testCuVectorNewFromSize(self):
vec = CuVector()
self.assertIsNotNone(vec)
self.assertEqual(0, vec.dim())
for i in range(10):
dim = 10 * i
vec = CuVector.new_from_size(dim)
self.assertIsNotNone(vec)
self.assertEqual(dim, vec.dim())
def testCuVectorSwap(self):
N = [2, 3, 5, 7, 13]
A = Vector(N).clone()
C = CuVector.new_from_size(5)
C.swap(A) #Swap *is* destructive
self.assertEqual(16.0, C.norm(2))
A = Vector()
C = CuVector.new_from_size(0)
C.swap(A)
self.assertEqual(0.0, C.norm(2))
def testCuVectorGetItem(self):
v = CuVector()
with self.assertRaises(IndexError):
v[0]
v = CuVector.new([3, 5, 7, 11, 13])
self.assertAlmostEqual(3.0, v[0])
self.assertAlmostEqual(7.0, v[2])
self.assertAlmostEqual(13.0, v[4])
with self.assertRaises(IndexError):
v[5]
def testCuVectorInverElements(self):
# Test that this doesnt crash
C = CuVector.new_from_size(0)
C.invert_elements()
C = CuVector.new_from_size(10)
C.set_randn()
C.invert_elements()
# Geometric series r = 1/2, a = 1/2
A = Vector([2, 4, 8, 16, 32, 64])
C = CuVector.new_from_size(len(A))
C.swap(A)
C.invert_elements()
f1 = C.sum()
self.assertAlmostEqual(0.5 * (1 - 0.5**len(A)) / (1 - 0.5), f1)
def testCuVectorCopyFromVec(self):
# Shouldnt crash
A = Vector()
C = CuVector.new_from_size(0)
C.copy_from_vec(A)
# What if dims not match?
# HARD-CRASH
# FIXME
# A = Vector.random(10)
# C = CuVector.new_from_size(0)
# C.CopyFromVec(A)
for i in range(10):
dim = 10 * i
A = Vector(dim)
A.set_randn_()
D = CuVector.new_from_size(dim)
D.copy_from_vec(A)
self.assertEqual(A.sum(), D.sum())
def testCuSubVector(self):
for iteration in range(10):
M1 = 1 + kaldi_math.rand() % 10
M2 = 1 + kaldi_math.rand() % 1
M3 = 1 + kaldi_math.rand() % 10
M = M1 + M2 + M3
m = kaldi_math.rand() % M2
vec = CuVector.new_from_size(M)
vec.set_randn()
subvec1 = CuSubVector(vec, M1, M2)
# subvec2 = vec.range(M1, M2)
f1, f2, f3 = vec[M1 + m], subvec1[m], subvec2[m]
self.assertEqual(f1, f2)
self.assertEqual(f3, f2)
def testCuVectorResize(self):
for i in range(10):
dim = 10 * i
vec = CuVector()
vec.resize(dim)
self.assertEqual(dim, vec.dim())
