def test_splitlist_1(self, mpi_mock): sequence = self._get_sequence(7) n = 1 sList = mpi_util._splitList(sequence, n) assert len(sList) == n for i in range(len(sequence)): assert np.all(sList[0][i] == sequence[i])
def test_splitlist_80(self, mpi_mock): sequence = self._get_sequence(160) n = 80 sList = mpi_util._splitList(sequence, n) assert len(sList) == n l0 = 2 for k in range(n): assert len(sList[k]) == l0 for i in range(l0): assert np.all(sList[k][i] == sequence[(k * l0) + i])
def test_splitlist_80(self, mpi_mock): sequence = self._get_sequence(160) n = 80 sList = mpi_util._splitList(sequence, n) assert len(sList) == n l0 = 2 for k in range(n): assert len(sList[k]) == l0 for i in range(l0): assert np.all(sList[k][i] == sequence[(k*l0)+i])
def test_splitlist_2(self, mpi_mock): sequence = self._get_sequence(7) n = 2 sList = mpi_util._splitList(sequence, n) assert len(sList) == n assert len(sList[0]) == 4 for i in range(4): assert np.all(sList[0][i] == sequence[0 + i]) assert len(sList[1]) == 3 for i in range(3): assert np.all(sList[1][i] == sequence[4 + i])
def test_splitlist_2(self, mpi_mock): sequence = self._get_sequence(7) n = 2 sList = mpi_util._splitList(sequence, n) assert len(sList) == n assert len(sList[0]) == 4 for i in range(4): assert np.all(sList[0][i] == sequence[0+i]) assert len(sList[1]) == 3 for i in range(3): assert np.all(sList[1][i] == sequence[4+i])
def test_splitlist_3(self, mpi_mock): sequence = self._get_sequence(7) n = 3 sList = mpi_util._splitList(sequence, n) assert len(sList) == n l0 = 2 assert len(sList[0]) == l0 for i in range(l0): assert np.all(sList[0][i] == sequence[0 + i]) l1 = 3 assert len(sList[1]) == l1 for i in range(l1): assert np.all(sList[1][i] == sequence[2 + i]) l2 = 2 assert len(sList[2]) == l2 for i in range(l2): assert np.all(sList[2][i] == sequence[5 + i])
def test_splitlist_3(self, mpi_mock): sequence = self._get_sequence(7) n = 3 sList = mpi_util._splitList(sequence, n) assert len(sList) == n l0 = 2 assert len(sList[0]) == l0 for i in range(l0): assert np.all(sList[0][i] == sequence[0+i]) l1 = 3 assert len(sList[1]) == l1 for i in range(l1): assert np.all(sList[1][i] == sequence[2+i]) l2 = 2 assert len(sList[2]) == l2 for i in range(l2): assert np.all(sList[2][i] == sequence[5+i])