Split test_arange into two tests (shape and stop)

These changes address the comments in #161, namely splitting the tests
and removing the extra underscore in the main function. test_arange is
now `test_arange_shape` and `test_arange_stop`, which are the two
fundamental ways of interacting with the `arange` API.