def test_generator_size():
my_generator_384 = plate_generator.generator(n=10, size=384)
for plate, int_label in my_generator_384:
assert plate.size == 384
assert plate.shape == (16, 24)
##
my_generator_1536 = plate_generator.generator(n=10, size=1536)
for plate, int_label in my_generator_1536:
assert plate.size == 1536
assert plate.shape == (32, 48)
def test_generator_n():
"""check it creates the correct number of plates"""
count = 0
N = 1000
my_generator = plate_generator.generator(n=N)
for i in my_generator:
count += 1
assert count == N
def test_generator_is_not_too_slow():
# 1500 plates per second minimum
time_limit = 10
n = 15000
my_generator = plate_generator.generator(n)
time_start = time()
for plate in my_generator:
_ = plate
time_end = time()
time_taken = abs(time_end - time_start)
print(time_taken)
assert time_taken <= time_limit
def test_generator_random_size():
count_384 = 0
count_1536 = 0
n = 100
my_generator = plate_generator.generator(n=n, size="either")
for plate, _ in my_generator:
plate_size = plate.size
if plate_size == 384:
count_384 += 1
if plate_size == 1536:
count_1536 += 1
assert count_384 > 0
assert count_1536 > 0
assert count_384 + count_1536 == n
def test_generator_select_size():
for output in plate_generator.generator(n=1000, size=384):
assert output.plate.size == 384
for output in plate_generator.generator(n=1000, size=1536):
assert output.plate.size == 1536
def test_generator():
my_generator = plate_generator.generator(n=10)
for plate, int_label in my_generator:
assert isinstance(plate, plate_generator.Plate)