File: iterator_test.py

package info (click to toggle)
keras-preprocessing 1.1.2-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 400 kB
  • sloc: python: 4,161; makefile: 11; sh: 10
file content (9 lines) | stat: -rw-r--r-- 304 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
 
from keras_preprocessing.image import iterator


def test_iterator_empty_directory():
    # Testing with different batch sizes
    for batch_size in [0, 32]:
        data_iterator = iterator.Iterator(0, batch_size, False, 0)
        ret = next(data_iterator.index_generator)
        assert ret.size == 0