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")