Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 63e06fe7 authored by hhakim's avatar hhakim
Browse files

Complete unit tests for pyfaust.lazylinop.LazyLinearOpKron/kron.

parent f6c005cd
No related branches found
No related tags found
No related merge requests found
...@@ -210,5 +210,18 @@ class TestLazyLinearOpFaust(unittest.TestCase): ...@@ -210,5 +210,18 @@ class TestLazyLinearOpFaust(unittest.TestCase):
self.assertTrue(pf.lazylinop.isLazyLinearOp(lcF)) self.assertTrue(pf.lazylinop.isLazyLinearOp(lcF))
self.assertEqual(cF.shape, lcF.shape) self.assertEqual(cF.shape, lcF.shape)
class TestLazyLinearOpFaustKron(TestLazyLinearOpFaust):
def setUp(self):
from pyfaust.lazylinop import kron as lkron
lop_A = LazyLinearOp.create(pf.rand(10, 15))
lop_B = LazyLinearOp.create(pf.rand(10, 15))
self.lop = lkron(lop_A, lop_B)
self.lopA = self.lop.toarray()
self.lop2 = LazyLinearOp.create(pf.rand(*self.lop.shape))
self.lop2A = self.lop2.toarray()
self.lop3 = LazyLinearOp.create(pf.rand(self.lop.shape[1], 10))
self.lop3A = self.lop3.toarray()
if '__main__' == __name__: if '__main__' == __name__:
unittest.main() unittest.main()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment