Added tests for scalar take and also added tests for using np.take as a function