13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
17
from bzrlib import tests
18
from bzrlib.tests import test_lsprof
18
from bzrlib.tests import features
21
21
class TestLSProf(tests.TestCaseInTempDir):
23
_test_needs_features = [test_lsprof.LSProfFeature]
23
_test_needs_features = [features.lsprof_feature]
25
25
def test_file(self):
26
out, err = self.run_bzr('--lsprof-file', 'output.callgrind', 'rocks')
26
out, err = self.run_bzr('--lsprof-file output.callgrind rocks')
27
27
self.assertNotContainsRe(out, 'Profile data written to')
28
28
self.assertContainsRe(err, 'Profile data written to')
30
30
def test_stdout(self):
31
out, err = self.run_bzr('--lsprof', 'rocks')
31
out, err = self.run_bzr('--lsprof rocks')
32
32
self.assertContainsRe(out, 'CallCount')
33
33
self.assertNotContainsRe(err, 'Profile data written to')