def test_translate_dlist(self): dct = {'a': 'something', 'b': 'other'} a = Aliases(dct) lst = [{'a': 3, 'b': 5}, {'a': 7, 'b': 8}] ret = a.translate(lst) expected = [{'something': 3, 'other': 5}, {'something': 7, 'other': 8}] self.assertListEqual(ret, expected)
def test_translate_list(self): dct = {'a': 'something'} a = Aliases(dct) lst = ['a'] ret = a.translate(lst) expected = ['something'] self.assertListEqual(ret, expected)
def test_translate_dict_vals(self): dct = {'a': 'something'} a = Aliases(dct) d = {'one': 'a'} ret = a.translate_dict_vals(d) expected = {'one': 'something'} self.assertDictEqual(ret, expected)
def test_translate_dict(self): dct = {'a': 'something'} a = Aliases(dct) d = {'a': 3} ret = a.translate(d) expected = {'something': 3} self.assertDictEqual(ret, expected)
def test_translate_str(self): dct = {'a': 'something'} a = Aliases(dct) s = 'a' ret = a.translate(s) expected = 'something' self.assertEqual(ret, expected)
def test_translate_dlist_variables(self): dct = { 'a': Variable('one', 'blah', 'mm'), 'b': Variable('two', 'blah2', 's') } a = Aliases(dct) lst = [{'a': 3, 'b': 5}, {'a': 7, 'b': 8}] ret = a.translate(lst) expected = [{ 'one [mm]': 3, 'two [s]': 5 }, { 'one [mm]': 7, 'two [s]': 8 }] self.assertListEqual(ret, expected)
def test_translate_boxlike(self): dct = {'a': 'something', 'b': 'other'} a = Aliases(dct) lst = [{ 'index': 0, 'independent': { 'c': 55 }, 'dependent': { 'a': 3, 'b': 5 } }, { 'index': 1, 'independent': { 'c': 66 }, 'dependent': { 'a': 7, 'b': 8 } }] ret = a.translate(lst) expected = [{ 'index': 0, 'independent': { 'c': 55 }, 'dependent': { 'something': 3, 'other': 5 } }, { 'index': 1, 'independent': { 'c': 66 }, 'dependent': { 'something': 7, 'other': 8 } }] self.assertListEqual(ret, expected)
def test_tabulate_store_units(self): t = Tabulator() store = Store() d = store.new('d', unit='mm', components=['x', 'y']) aliases = Aliases({'d': d}) translated = aliases.translate(get_lst2_b()) box = Box(translated) index = ['a', 'b'] columns = d.label values = d pt = t.tabulate(box=box, values=values, columns=columns, index=index, store=store) expected = '''d: x [mm] y [mm] a b 1 1 12 30 2 13 31 2 1 16 34 2 19 37''' self.assertEqual(str(pt), expected)
def test_tabulate_translated(self): dct = { 'a': Variable('one', 'blah', 'mm'), 'b': Variable('two', 'blah2', 's'), 'c': Variable('three', 'blah2', 'N') } a = Aliases(dct) t = Tabulator() box = Box(get_lst()) index = ['a', 'b'] columns = ['c'] values = 'd' pt = t.tabulate(box=box, values=values, columns=columns, index=index, aliases=a) expected = 'three [N] 1 2\none [mm] two [s] \n1 1 11 14\n 2 13 12\n2 1 15 16\n 2 17 18' self.assertEqual(str(pt), expected)
def test_init(self): dct = {'a': 'something'} a = Aliases(dct) self.assertDictEqual(a, dct)