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)