def test_cut_tiles_squared(self): arr = (rand(100, 100, 3) * 255).astype('uint8') image = Image.fromarray(arr).convert('RGBA') tiles = list(Tiler.cut_tiles(image, (10, 10))) assert len(tiles) == 100 last_tile = tiles[-1][0] assert last_tile.size == (10, 10) last_coords = tiles[-1][1] assert last_coords == (9, 9) assert (asarray(last_tile) == asarray(image)[-10:, -10:]).all()
def test_cut_tiles_incomplete(self): arr = (rand(95, 45, 3) * 255).astype('uint8') image = Image.fromarray(arr).convert('RGBA') tiles = list(Tiler.cut_tiles(image, (10, 10))) assert len(tiles) == 50 sec_last_tile = tiles[-2][0] assert sec_last_tile.size == (5, 10) sec_last_coords = tiles[-2][1] assert sec_last_coords == (4, 8) last_tile = tiles[-1][0] assert last_tile.size == (5, 5) last_coords = tiles[-1][1] assert last_coords == (4, 9) assert (asarray(last_tile) == asarray(image)[-5:, -5:]).all()