def test_get_matching_dim_error1(self):
dims = aobj.Dims((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_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_index1(self):
dims = aobj.Dims((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.Dims((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_contains1(self):
dims = (dim.DimSweep("a", 3), dim.DimSweep("b", 3),
dim.DimSweep("c", 3))
dims = aobj.Dims(dims)
for d in dims:
self.assertTrue(d in dims)