def test_args_to_iterable_mixed_strings(): gpis = [1, 2, 3] lons = [2, 3, 4] lats = 1 arg1 = "test" gpis_, lons_, lats_, args = args_to_iterable(gpis, lons, lats, arg1) assert gpis_ == gpis assert lons_ == lons assert lats_ == [lats] assert args == [arg1]
def test_args_to_iterable_mixed(): gpis = [1, 2, 3] lons = [2, 3, 4] lats = 1 arg1 = 1 gpis_, lons_, lats_, args = args_to_iterable(gpis, lons, lats, arg1) assert gpis_ == gpis assert lons_ == lons assert lats_ == [lats] assert args == [arg1]
def test_args_to_iterable_non_iterables(): gpis = 1 lons = 1 lats = 1 arg1 = 1 arg2 = 2 arg3 = 3 gpis_, lons_, lats_, args = args_to_iterable(gpis, lons, lats, arg1, arg2, arg3, n=3) assert gpis_ == [gpis] assert lons_ == [lons] assert lats_ == [lats] assert args == ([arg1], [arg2], [arg3])
def test_args_to_iterable_non_iterables(): gpis = 1 lons = 1 lats = 1 arg1 = 1 arg2 = 2 arg3 = 3 gpis_, lons_, lats_, args = args_to_iterable(gpis, lons, lats, arg1, arg2, arg3, n=3) assert gpis_ == [gpis] assert lons_ == [lons] assert lats_ == [lats] assert args == ([arg1], [arg2], [arg3])
def test_args_to_iterable_n3(): gpis = [1, 2, 3] lons = [2, 3, 4] lats = [3, 4, 5] arg1 = [1, 1, 1] arg2 = [1, 1, 1] gpis_, lons_, lats_, args = args_to_iterable( gpis, lons, lats, arg1, arg2, n=3 ) assert gpis_ == gpis assert lons_ == lons assert lats_ == lats assert args == (arg1, arg2) zipped_should = [(1, 2, 3, 1, 1), (2, 3, 4, 1, 1), (3, 4, 5, 1, 1)] for i, t in enumerate(zip(gpis_, lons_, lats_, *args)): assert zipped_should[i] == t
def test_args_to_iterable_n3(): gpis = [1, 2, 3] lons = [2, 3, 4] lats = [3, 4, 5] arg1 = [1, 1, 1] arg2 = [1, 1, 1] gpis_, lons_, lats_, args = args_to_iterable(gpis, lons, lats, arg1, arg2, n=3) assert gpis_ == gpis assert lons_ == lons assert lats_ == lats assert args == (arg1, arg2) zipped_should = [(1, 2, 3, 1, 1), (2, 3, 4, 1, 1), (3, 4, 5, 1, 1)] for i, t in enumerate(zip(gpis_, lons_, lats_, *args)): assert zipped_should[i] == t