def test_flatten_nested(self): flatArray = [i for i in APIUtils.flatten([[1, 2], 3, [4, [5]]])] assert flatArray == [1, 2, 3, 4, 5]
def test_flatten_already_flat(self): flatArray = [i for i in APIUtils.flatten([1, 2, 3, 4, 5])] assert flatArray == [1, 2, 3, 4, 5]