def test__flatten_array_list(self):
"""_flatten_array return a dict formed from the input list"""
# Setup
nested = [
[1, 0, 0],
[0, 1, 0],
[0, 0, 1]
]
expected_result = {
'0__0': 1,
'0__1': 0,
'0__2': 0,
'1__0': 0,
'1__1': 1,
'1__2': 0,
'2__0': 0,
'2__1': 0,
'2__2': 1
}
# Run
result = Modeler._flatten_array(nested)
# Check
assert result == expected_result
def test__flatten_array(self):
"""Test get flatten array"""
# Run
nested = [['foo', 'bar'], 'tar']
result = Modeler._flatten_array(nested, prefix='test')
# Asserts
expected = {'test__0__0': 'foo', 'test__0__1': 'bar', 'test__1': 'tar'}
assert result == expected
