def test_DKG_array2(self):
store = {'array': np.array([[1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12]])}
result = outputting.dictKeyGen(store)
correct_result = (collections.OrderedDict([('array', np.array([[0, 0], [1, 0], [0, 1], [1, 1], [0, 2], [1, 2], [0, 3], [1, 3], [0, 4], [1, 4],
[0, 5], [1, 5]]))]),
1)
assert (result[0]['array'] == correct_result[0]['array']).all()
assert result[1] == correct_result[1]
def test_DKG_dict3(self):
store = {'dict': {1: {3: "a"}, 2: "b"}}
result = outputting.dictKeyGen(store)
correct_result = (collections.OrderedDict([('dict', collections.OrderedDict([(1, collections.OrderedDict([(3, None)])),
(2, None)]))]),
1)
assert result[0]['dict'] == correct_result[0]['dict']
assert result[1] == correct_result[1]
def test_DKG_array3(self):
store = {'array': np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12]])}
result = outputting.dictKeyGen(store)
correct_result = (collections.OrderedDict([('array', np.array([[0, 0], [1, 0], [2, 0], [3, 0], [4, 0], [5, 0], [0, 1], [1, 1], [2, 1], [3, 1],
[4, 1], [5, 1]]))]),
1)
assert (result[0]['array'] == correct_result[0]['array']).all()
assert result[1] == correct_result[1]
def test_DKG_dict2(self):
store = {'dict': {1: [1, 2, 3], 2: [[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11], [12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22]]}}
result = outputting.dictKeyGen(store, abridge=True)
correct_result = (collections.OrderedDict([('dict', collections.OrderedDict([(1, np.array([[0], [1], [2]])),
(2, None)]))]),
1)
assert (result[0]['dict'][1] == correct_result[0]['dict'][1]).all()
assert result[0]['dict'][2] == correct_result[0]['dict'][2]
assert result[1] == correct_result[1]
def test_DKG_array(self):
store = {'array': np.array([1, 2, 3])}
result = outputting.dictKeyGen(store)
correct_result = (collections.OrderedDict([('array', np.array([[0], [1], [2]]))]), 1)
assert (result[0]['array'] == correct_result[0]['array']).all()
assert result[1] == correct_result[1]
def test_DKG_num(self):
store = {'num': 23.6}
result = outputting.dictKeyGen(store)
correct_result = (collections.OrderedDict([('num', None)]), 1)
assert result[0] == correct_result[0]
assert result[1] == correct_result[1]
def test_DKG_list1(self):
store = {'list': [1, 2, 3, 4, 5, 6]}
result = outputting.dictKeyGen(store)
correct_result = (collections.OrderedDict([('list', np.array([[0], [1], [2], [3], [4], [5]]))]), 1)
assert (result[0]['list'] == correct_result[0]['list']).all()
assert result[1] == correct_result[1]
def test_DKG_string(self):
store = {'string': 'string'}
result = outputting.dictKeyGen(store)
correct_result = (collections.OrderedDict([('string', None)]), 1)
assert result[0] == correct_result[0]
assert result[1] == correct_result[1]
def test_DKG_none(self):
store = {}
result = outputting.dictKeyGen(store)
correct_result = (collections.OrderedDict(), None)
assert result[0] == correct_result[0]
assert result[1] == correct_result[1]