Ok .. I successfully read into unittesting of python and added your unit tests for private key addresses pubkeys and wif format.
unit test can be executed via
$ python2 -m unittest genbtskey
.....
----------------------------------------------------------------------
Ran 5 tests in 1.893s
OK
I also added the option to have a single argument .. a WIF key .. and convert that to the corresponding address/pubkey ..