import foo
import unittest

class TestFoo(unittest.TestCase):

    def testFoo(self):
        self.assertEquals(foo.foo(), 6, 'foo() == 6')

if __name__ == '__main__':
    unittest.main()