def test_vectors_nonstr(self):
b = Box(get_lst3())
keys = ['d', 'e']
vec0, vec1, labels = b.vectors(keys, labels='dict')
expected_vec0 = [[12, 30], [13, 31]]
expected_vec1 = [[1, 2], [1, 2]]
expected_labels = [{'a': 1, 'b': 1}, {'a': 1, 'b': 2}]
self.assertListEqual(expected_vec0, vec0)
self.assertListEqual(expected_vec1, vec1)
self.assertListEqual(expected_labels, labels)
def test_vectors_uncombined(self):
b = Box(get_lst5())
keys = ['b', 'e']
vec0, vec1, labels = b.vectors(keys, combine=False)
expected_vec0 = [[1, 2], [1, 2]]
expected_vec1 = [[12, 30], [13, 31]]
expected_labels = ['a=1', 'a=1']
self.assertListEqual(expected_vec0, vec0)
self.assertListEqual(expected_vec1, vec1)
self.assertListEqual(expected_labels, labels)
def test_vectors_indep_keys(self):
b = Box(get_lst4())
keys = ['c']
c, a, b = b.vectors(keys, indep_keys=['a', 'b'])
expected_c = [4, 5, 6, 7]
expected_a = [1, 1, 2, 2]
expected_b = [1, 2, 1, 2]
self.assertListEqual(expected_c, c)
self.assertListEqual(expected_a, a)
self.assertListEqual(expected_b, b)
def test_vectors(self):
b = Box(get_lst3())
keys = ['d', 'e']
vec0, vec1, labels = b.vectors(keys)
expected_vec0 = [[12, 30], [13, 31]]
expected_vec1 = [[1, 2], [1, 2]]
expected_labels = ['a=1, b=1', 'a=1, b=2']
self.assertListEqual(expected_vec0, vec0)
self.assertListEqual(expected_vec1, vec1)
self.assertListEqual(expected_labels, labels)
def test_vectors_scalar(self):
b = Box(get_lst4())
keys = ['c']
c, labels = b.vectors(keys, labels='dict')
expected_c = [4, 5, 6, 7]
expected_labels = [{
'a': 1,
'b': 1
}, {
'a': 1,
'b': 2
}, {
'a': 2,
'b': 1
}, {
'a': 2,
'b': 2
}]
self.assertListEqual(expected_c, c)
self.assertListEqual(expected_labels, labels)