Change doctests from print to assert statements

Address issue #109

The doctest tests in stats.py and discrete.py were changed from print
statements to assert statements. In test.py that needed a simple change to the
dimensions used in the assert statement. All three files had slight style
modifications to be in line with PEP8 standards.