Example #1
0
 def test_1(self):
     out = dset.convert_matrices_to_elements(self.db)
     self.assertTrue("V" in out)
     self.assertTrue("S11" in out)
     self.assertTrue("S12" in out)
     self.assertTrue("S21" in out)
     self.assertTrue("S22" in out)
     self.assertTrue("Y" in out)
Example #2
0
 def test_1(self):
     out = dset.convert_matrices_to_elements(self.db)
     self.assertTrue("V" in out)
     self.assertTrue("S11" in out)
     self.assertTrue("S12" in out)
     self.assertTrue("S21" in out)
     self.assertTrue("S22" in out)
     self.assertTrue("Y" in out)
Example #3
0
 def test_2(self):
     def formatelement(varname, i, j):
         return "%s%s%s" % (varname, i, j)
     out = dset.convert_matrices_to_elements(self.db, formatelement)
     self.assertTrue("V" in out)
     self.assertTrue("S11" in out)
     self.assertTrue("S12" in out)
     self.assertTrue("S21" in out)
     self.assertTrue("S22" in out)
     self.assertTrue("Y" in out)
Example #4
0
    def test_2(self):
        def formatelement(varname, i, j):
            return "%s%s%s" % (varname, i, j)

        out = dset.convert_matrices_to_elements(self.db, formatelement)
        self.assertTrue("V" in out)
        self.assertTrue("S11" in out)
        self.assertTrue("S12" in out)
        self.assertTrue("S21" in out)
        self.assertTrue("S22" in out)
        self.assertTrue("Y" in out)
Example #5
0
 def test_1(self):
     db = DataBlock()
     Sdims = (DimRep("freq", [1, 2]) ,DimRep("r", [1]), DimMatrix_i("i", 2), DimMatrix_j("j", 2))
     db.S = hfarray(np.array([[11, 12], [21, 22]])[np.newaxis, np.newaxis, :, :] * np.array([[10], [20]])[..., np.newaxis, np.newaxis], dims=Sdims)
     db.V = hfarray([1.23], Sdims[1:2])
     db.Y = hfarray([1.23], (DimRep("k", [1]),))
     out = dset.subset_datablock_by_dims(dset.convert_matrices_to_elements(db), Sdims[:-2])
     self.assertTrue("V" in out)
     self.assertTrue("S11" in out)
     self.assertTrue("S12" in out)
     self.assertTrue("S21" in out)
     self.assertTrue("S22" in out)
     self.assertFalse("Y" in out)
Example #6
0
 def test_1(self):
     db = DataBlock()
     Sdims = (DimRep("freq", [1, 2]), DimRep("r", [1]), DimMatrix_i("i", 2),
              DimMatrix_j("j", 2))
     db.S = hfarray(
         np.array([[11, 12], [21, 22]])[np.newaxis, np.newaxis, :, :] *
         np.array([[10], [20]])[..., np.newaxis, np.newaxis],
         dims=Sdims)
     db.V = hfarray([1.23], Sdims[1:2])
     db.Y = hfarray([1.23], (DimRep("k", [1]), ))
     out = dset.subset_datablock_by_dims(
         dset.convert_matrices_to_elements(db), Sdims[:-2])
     self.assertTrue("V" in out)
     self.assertTrue("S11" in out)
     self.assertTrue("S12" in out)
     self.assertTrue("S21" in out)
     self.assertTrue("S22" in out)
     self.assertFalse("Y" in out)
Example #7
0
 def test1(self):
     res = list(
         dset.yield_dim_consistent_datablocks(
             dset.convert_matrices_to_elements(self.db)))
     self.assertTrue(len(res) == 2)
Example #8
0
 def test1(self):
     res = list(dset.yield_dim_consistent_datablocks(dset.convert_matrices_to_elements(self.db)))
     self.assertTrue(len(res) == 2)