fix #1171 -- support `setup.py test`

Signed-off-by: Chris “Kwpolska” Warrick <kwpolska@gmail.com>