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)