import numpy as np

values = np.arange(7)
parts = np.array_split(values, 3)
print([part.tolist() for part in parts])
try:
    np.split(values, 3)
except ValueError:
    print("Rejected unequal split")