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