def _test_index_iterator_serialization_no_shuffle():
index_list = [1, 3, 5, 10]
ii = IndexIterator(index_list, shuffle=False, num=2)
indices1 = ii.get_next_indices(3) # NOQA
indices2 = ii.get_next_indices(6) # NOQA
indices3 = ii.__next__() # NOQA
assert len(ii.current_index_list) == len(index_list)
assert numpy.array_equal(ii.current_index_list, numpy.asarray(index_list))
assert ii.current_pos == (3 + 6) % len(index_list) + 2
target = dict()
ii.serialize(DummySerializer(target))
ii = IndexIterator(index_list, shuffle=False, num=2)
ii.serialize(DummyDeserializer(target))
assert len(ii.current_index_list) == len(index_list)
assert numpy.array_equal(ii.current_index_list, numpy.asarray(index_list))
assert ii.current_pos == (3 + 6) % len(index_list) + 2
def _test_index_iterator_serialization_no_shuffle():
index_list = [1, 3, 5, 10]
ii = IndexIterator(index_list, shuffle=False, num=2)
indices1 = ii.get_next_indices(3)
indices2 = ii.get_next_indices(6)
indices3 = ii.__next__()
assert len(ii.current_index_list) == len(index_list)
assert numpy.array_equal(ii.current_index_list, numpy.asarray(index_list))
assert ii.current_pos == (3 + 6) % len(index_list) + 2
target = dict()
ii.serialize(DummySerializer(target))
ii = IndexIterator(index_list, shuffle=False, num=2)
ii.serialize(DummyDeserializer(target))
assert len(ii.current_index_list) == len(index_list)
assert numpy.array_equal(ii.current_index_list, numpy.asarray(index_list))
assert ii.current_pos == (3 + 6) % len(index_list) + 2
def _test_index_iterator_serialization_with_shuffle():
index_list = [1, 3, 5, 10]
ii = IndexIterator(index_list, shuffle=True, num=2)
indices1 = ii.get_next_indices(3) # NOQA
indices2 = ii.get_next_indices(6) # NOQA
indices3 = ii.__next__() # NOQA
assert len(ii.current_index_list) == len(index_list)
for index in ii.current_index_list:
assert index in index_list
assert ii.current_pos == (3 + 6) % len(index_list) + 2
target = dict()
ii.serialize(DummySerializer(target))
current_index_list_orig = ii.current_index_list
ii = IndexIterator(index_list, shuffle=True, num=2)
ii.serialize(DummyDeserializer(target))
assert numpy.array_equal(ii.current_index_list, current_index_list_orig)
assert ii.current_pos == (3 + 6) % len(index_list) + 2
def _test_index_iterator_serialization_with_shuffle():
index_list = [1, 3, 5, 10]
ii = IndexIterator(index_list, shuffle=True, num=2)
indices1 = ii.get_next_indices(3)
indices2 = ii.get_next_indices(6)
indices3 = ii.__next__()
assert len(ii.current_index_list) == len(index_list)
for index in ii.current_index_list:
assert index in index_list
assert ii.current_pos == (3 + 6) % len(index_list) + 2
target = dict()
ii.serialize(DummySerializer(target))
current_index_list_orig = ii.current_index_list
ii = IndexIterator(index_list, shuffle=True, num=2)
ii.serialize(DummyDeserializer(target))
assert numpy.array_equal(ii.current_index_list, current_index_list_orig)
assert ii.current_pos == (3 + 6) % len(index_list) + 2