From 0b6ac7bdec335ffab6872aed96f52bb05d4a881d Mon Sep 17 00:00:00 2001 From: Anton Beloglazov Date: Mon, 22 Oct 2012 16:46:34 +1100 Subject: [PATCH] Fixed tests --- tests/locals/test_collector.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/locals/test_collector.py b/tests/locals/test_collector.py index 54d0b38..854a12a 100644 --- a/tests/locals/test_collector.py +++ b/tests/locals/test_collector.py @@ -484,11 +484,11 @@ class Collector(TestCase): with MockTransaction: context = mock('context') f = mock('file') - when(context).__enter__().and_return(f)#.once() + expect(context).__enter__().and_return(f).once() when(context).__exit__.and_return(True) - when(collector).open('/proc/stat', 'r').and_return(context)#.once() - when(f).readline().and_return( - '1 ' + ' '.join([str(v) for v in x]) + ' 2 3')#.once() + expect(collector).open('/proc/stat', 'r').and_return(context).once() + expect(f).readline().and_return( + '1 ' + ' '.join([str(v) for v in x]) + ' 2 3').once() assert collector.get_host_cpu_time() == (float(sum(x)), float(sum(x[0:3])))