def test_get_matching_dim_error1(self):
dims = aobj.DimsList((dim.DimSweep("a", 3), dim.DimSweep("b", 3),
dim.DimSweep("c", 3)))
self.assertRaises(KeyError, dims.get_matching_dim, dim.DimRep("c", 4))
self.assertRaises(KeyError, dims.get_matching_dim,
dim.DimSweep("d", 4))
self.assertRaises(KeyError, dims.get_matching_dim, "a")
def test_contains1(self):
dims = (dim.DimSweep("a", 3), dim.DimSweep("b", 3),
dim.DimSweep("c", 3))
dims = aobj.DimsList(dims)
for d in dims:
self.assertTrue(d in dims)
self.assertFalse("a" in dims)
def test_get_new_anon_2(self):
dims = (
dim.DimSweep("ANON1", 3),
dim.DimSweep("ANON2", 3),
)
anon = aobj.get_new_anonymous_dim(dims, [1, 2, 3])
self.assertIsInstance(anon, aobj.DimAnonymous)
self.assertEqual(anon.name, "ANON1")
def setUp(self):
(ai, bi, ci) = (dim.DimSweep("ai",
2), dim.DimRep("bi",
3), dim.DimSweep("ci", 4))
self.ai = ai
self.bi = bi
self.ci = ci
self.a = aobj.hfarray(ai)
self.b = aobj.hfarray(bi) * 10
self.c = aobj.hfarray(ci) * 100
self.abc = self.a + self.b + self.c
def test_matching_index1(self):
dims = aobj.DimsList((dim.DimSweep("a", 3), dim.DimSweep("b", 3),
dim.DimSweep("c", 3)))
self.assertEqual(dims.matching_index(dim.DimSweep("a", 4)), 0)
self.assertEqual(dims.matching_index(dim.DimSweep("b", 2)), 1)
self.assertEqual(dims.matching_index(dim.DimSweep("c", 4)), 2)
def test_get_matching_dim1(self):
dims = aobj.DimsList((dim.DimSweep("a", 3), dim.DimSweep("b", 3),
dim.DimSweep("c", 3)))
self.assertEqual(dims.get_matching_dim(dim.DimSweep("a", 4)), dims[0])
self.assertEqual(dims.get_matching_dim(dim.DimSweep("b", 2)), dims[1])
self.assertEqual(dims.get_matching_dim(dim.DimSweep("c", 4)), dims[2])
def test_containing_1(self):
dims = aobj.Dims((dim.DimSweep("a", 3), dim.DimSweep("b", 3),
dim.DimSweep("c", 3)))
self.assertTrue("c" in dims)
self.assertFalse("d" in dims)
def test_matching_index_error1(self):
dims = aobj.Dims((dim.DimSweep("a", 3), dim.DimSweep("b", 3),
dim.DimSweep("c", 3)))
self.assertRaises(KeyError, dims.matching_index, dim.DimRep("c", 4))
self.assertRaises(KeyError, dims.matching_index, dim.DimSweep("d", 4))
self.assertRaises(KeyError, dims.matching_index, "a")
