def test_chunk_sizes(faces, Nx, Ny, rot, exp_tNX, exp_tNY): if _is_connect(faces, rotated=rot): tNy, tNx = chunk_sizes(faces, [Nx], [Ny], rotated=rot) assert tNy == exp_tNY assert tNx == exp_tNX else: with pytest.raises(ValueError): tNy, tNx = chunk_sizes(faces, [Nx], [Ny], rotated=rot) assert tNy == exp_tNY assert tNx == exp_tNX
def test_make_chunks(faces, rot, NX, NY, expCX, expCY, epx, epy, epax, epay): if rot: fs = [k for k in faces if k in _np.arange(7, 13)] else: fs = [k for k in faces if k in _np.arange(6)] tNy, tNx = chunk_sizes(faces, [Nx], [Ny], rotated=rot) delNX = 0 delNY = 0 afs = [] if 6 in faces: acnrot_fs = [k for k in faces if k in _np.array([2, 5])] acrot_fs = [k for k in faces if k in _np.array([7, 10])] if rot: delNX = int(Nx / 2) afs = acrot_fs else: delNY = int(Ny / 2) afs = acnrot_fs tNy = tNy + delNY tNx = tNx + delNX Nxc = _np.arange(0, tNx + 1, Nx) Nyc = _np.arange(0, tNy + 1, Ny) xChunk, yChunk = make_chunks(Nxc, Nyc) assert xChunk == expCX assert yChunk == expCY py, px, pyarc, pxarc = pos_chunks(fs, afs, yChunk, xChunk) assert epy == py assert epx == px assert epay == pyarc assert epax == pxarc