无法在unittest中导入类

这可能是一个初级问题,但我真的很紧张。

#!/usr/bin/env python

class Foo():
  def __init__(self):
    self.do_something()

  def do_something(self):
    print "foo"

def main():
  Foo()

if __name__ == '__main__':
  main()

脚本运行良好:

$ python foo.py 
foo

我想在unittest中测试函数“do\u something”,我有以下代码:

#!/usr/bin/env python

import unittest
import foo
from foo import *

class TestFoo(unittest.TestCase):
  def test_foo(self):
    Foo()

def main():
  unittest.main()

if __name__ == "__main__":
  main()

$ python pyunit/foo.py
E
======================================================================
ERROR: test_foo (__main__.TestFoo)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "pyunit/foo.py", line 9, in test_foo
    Foo()
NameError: global name 'Foo' is not defined

----------------------------------------------------------------------
Ran 1 test in 0.000s

FAILED (errors=1)

我的项目结构:

$ tree 
.
├── foo.py
└── pyunit
    └── foo.py

inspect dir() ,Python调试器等等,但我什么都做不到。

虽然 import foo 和 from foo import * main 从该文件导入,类本身 Foo

最终,我的目标是为类中的函数编写单元测试 傅 .

我做错了什么?