def test_bm(self): for num_layers in TEST_LAYERS: bm = BMNumpy(num_layers=num_layers, basis='sm') reconstructed_bm = parallel_nullification(bm) self.assertAllClose(bm.matrix, reconstructed_bm.matrix)
def test_tm(self): for units in TEST_DIMENSIONS: tm = TMNumpy(units=units, basis='sm') reconstructed_tm = parallel_nullification(tm) self.assertAllClose(tm.matrix, reconstructed_tm.matrix)