您可以在每个源文件的顶部插入以下行，强制 matplotlib 使用“Agg”后端（不会打开任何窗口）：
import matplotlib
matplotlib.use('Agg')
这是一个一行命令的shell指令，它会在将输出流传输到Python解释器执行之前，在
my_script.py文件的顶部
动态地插入这些行，而不会修改磁盘上的文件。
~$ sed "1i import matplotlib\nmatplotlib.use('Agg')\n" my_script.py | python
您应该能够使用
subprocess 进行相应的调用，例如：
p1 = sb.Popen(["sed", "1i import matplotlib\nmatplotlib.use('Agg')\n", fl],
stdout=sb.PIPE)
exit_cond = sb.call(["python"], stdin=p1.stdout)
您可以通过将
stdout=和
stderr=参数传递给
sb.call()来捕获脚本的
stderr和
stdout。当然，这只适用于具有
sed实用程序的Unix环境。
更新
这实际上是一个非常有趣的问题。我想了一下，我认为这是一种更优雅的解决方案（虽然仍然有点像黑客）：
import sys
import os
import glob
from contextlib import contextmanager
import traceback
set_backend = "import matplotlib\nmatplotlib.use('Agg')\n"
@contextmanager
def redirected_output(new_stdout=None, new_stderr=None):
save_stdout = sys.stdout
save_stderr = sys.stderr
if new_stdout is not None:
sys.stdout = new_stdout
if new_stderr is not None:
sys.stderr = new_stderr
try:
yield None
finally:
sys.stdout = save_stdout
sys.stderr = save_stderr
def run_exectests(test_dir, log_path='exectests.log'):
test_files = glob.glob(os.path.join(test_dir, '*.py'))
test_files.sort()
passed = []
failed = []
with open(log_path, 'w') as f:
with redirected_output(new_stdout=f, new_stderr=f):
for fname in test_files:
print(">> Executing '%s'" % fname)
try:
code = compile(set_backend + open(fname, 'r').read(),
fname, 'exec')
exec(code, {'__name__':'__main__'}, {})
passed.append(fname)
except:
traceback.print_exc()
failed.append(fname)
pass
print ">> Passed %i/%i tests: " %(len(passed), len(test_files))
print "Passed: " + ', '.join(passed)
print "Failed: " + ', '.join(failed)
print "See %s for details" % log_path
return passed, failed
if __name__ == '__main__':
run_exectests(*sys.argv[1:])
概念上，这与我的先前解决方案非常相似——它通过将测试脚本作为字符串读入，并在其前面添加一些行来导入matplotlib并将后端设置为非交互式的方式工作。然后将该字符串编译为Python字节码，再执行。其主要优点是应该是与平台无关的，因为不需要使用sed。
使用globals和{'__name__':'__main__'}技巧是必要的，如果像我一样倾向于像这样编写脚本：
def run_me():
...
if __name__ == '__main__':
run_me()
需要考虑以下几点：
- 如果您尝试在已经导入matplotlib并设置交互式后端的ipython会话中运行此函数，则“set_backend”技巧将无效，您仍将看到弹出的图形。最简单的方法是直接从shell（
~$ python exectests.py testdir/ logfile.log）或者在没有为matplotlib设置交互式后端的（i）python会话中直接运行它。如果您在ipython会话内的不同子进程中运行它，它也应该可以工作。
- 我使用这个答案中的“contextmanager”技巧将stdin和stdout重定向到日志文件。请注意，这不是线程安全的，但我认为脚本打开子进程相当少见。
