test_main.py 文件源码

def test_main_01_from_project(self):
        """Test the :func:`psyplot.__main__.main` function"""
        if not six.PY2:
            with self.assertRaisesRegex(ValueError, 'filename'):
                main.main(['-o', 'test.pdf'])
        sp, fname1 = self._create_and_save_test_project()
        fname2 = tempfile.NamedTemporaryFile(
                    suffix='.pdf', prefix='test_psyplot_').name
        self._created_files.add(fname2)
        sp.save_project(fname1, use_rel_paths=False)
        psy.close('all')
        if six.PY2:
            main.main(['-p', fname1, '-o', fname2])
        else:
            with self.assertWarnsRegex(UserWarning, 'ignored'):
                main.main(['-p', fname1, '-o', fname2, '-n', 't2m'])
        self.assertTrue(osp.exists(fname2), msg='Missing ' + fname2)
        self.assertEqual(len(psy.gcp(True)), 4)