Numpy Generate Square Wave