def test_1(self): d = DataBlock() dims = (DimSweep("Freq[Hz]", 3), ) d.P = hfarray([1, 2, 3], dims=dims) d.L = hfarray([10, 20, 30], dims=dims) self.assertEqual(d.allvarnames, ["Freq[Hz]", "P", "L"]) d.rename("P", "H") self.assertEqual(d.allvarnames, ["Freq[Hz]", "H", "L"]) d.rename("Freq[Hz]", "f") self.assertEqual(d.allvarnames, ["f", "H", "L"]) d.rename("foo", "bar") self.assertEqual(d.allvarnames, ["f", "H", "L"])
def test_1(self): d = DataBlock() dims = (DimSweep("Freq[Hz]", 3),) d.P = hfarray([1, 2, 3], dims=dims) d.L = hfarray([10, 20, 30], dims=dims) self.assertEqual(d.allvarnames, ["Freq[Hz]", "P", "L"]) d.rename("P", "H") self.assertEqual(d.allvarnames, ["Freq[Hz]", "H", "L"]) d.rename("Freq[Hz]", "f") self.assertEqual(d.allvarnames, ["f", "H", "L"]) d.rename("foo", "bar") self.assertEqual(d.allvarnames, ["f", "H", "L"])
def test_2(self): d = DataBlock() fi = DimSweep("Freq[Hz]", 3) gi = DimSweep("P[W]", 2) dims = (fi, ) dims2 = (fi, gi) dims3 = (gi, ) d.P = hfarray([1, 2, 3], dims=dims) d.Q = hfarray([3, 2], dims=dims3) d.L = hfarray([[10, 20, 30], [11, 21, 31]], dims=dims2) self.assertEqual(d.allvarnames, ["Freq[Hz]", "P[W]", "P", "Q", "L"]) d.rename("P", "H") self.assertEqual(d.allvarnames, ["Freq[Hz]", "P[W]", "H", "Q", "L"]) d.rename("Freq[Hz]", "f") self.assertEqual(d.allvarnames, ["f", "P[W]", "H", "Q", "L"]) d.rename("foo", "bar") self.assertEqual(d.allvarnames, ["f", "P[W]", "H", "Q", "L"])
def test_2(self): d = DataBlock() fi = DimSweep("Freq[Hz]", 3) gi = DimSweep("P[W]", 2) dims = (fi,) dims2 = (fi, gi) dims3 = (gi,) d.P = hfarray([1, 2, 3], dims=dims) d.Q = hfarray([3, 2], dims=dims3) d.L = hfarray([[10, 20, 30], [11, 21, 31]], dims=dims2) self.assertEqual(d.allvarnames, ["Freq[Hz]", "P[W]", "P", "Q", "L"]) d.rename("P", "H") self.assertEqual(d.allvarnames, ["Freq[Hz]", "P[W]", "H", "Q", "L"]) d.rename("Freq[Hz]", "f") self.assertEqual(d.allvarnames, ["f", "P[W]", "H", "Q", "L"]) d.rename("foo", "bar") self.assertEqual(d.allvarnames, ["f", "P[W]", "H", "Q", "L"])