test.py

from pymath import *
import nose

def test_add_integers():
    assert add(5, 3) == 8

def test_add_floats():
    assert add(1.5, 2.5) == 4

def test_divide_integers():
    assert divide(2, 10) == 0.2

if __name__ == '__main__':
    nose.run()