Update tests for progress bar context

florianeinfalt · Feb 18, 2017 · ec6f2ed · ec6f2ed
1 parent 92ec599
commit ec6f2ed
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 9 deletions.
diff --git a/nukecontexts/__init__.py b/nukecontexts/__init__.py
@@ -8,7 +8,7 @@
 TESTING = False
 try:
     TESTING = os.environ['NON_PRODUCTION_CONTEXT']
-    logger = None
+    logger = logging.getLogger()
     sentry = None
 except:
     if platform.system() == 'Darwin':

diff --git a/tests/test_contexts.py b/tests/test_contexts.py
@@ -5,31 +5,31 @@
 def test_enabled(node):
     node['disable'].setValue(True)
     assert node['disable'].value() == True
-    with ctx.enabled(node, verbose=True):
+    with ctx.enabled(node):
         assert node['disable'].value() == False
     assert node['disable'].value() == True
     node['disable'].setValue(False)
 
 
 def test_disabled(node):
     assert node['disable'].value() == False
-    with ctx.disabled(node, verbose=True):
+    with ctx.disabled(node):
         assert node['disable'].value() == True
     assert node['disable'].value() == False
 
 
 def test_set_attr_errors(node):
     # Test invalid node error
     with pytest.raises(ctx.NukeContextError):
-        with ctx.set_attr(None, 'invalid_attr', True, verbose=True):
+        with ctx.set_attr(None, 'invalid_attr', True):
             print 'should not print'
     # Test invalid attribute error
     with pytest.raises(ctx.NukeContextError):
-        with ctx.set_attr(node, 'invalid_attr', True, verbose=True):
+        with ctx.set_attr(node, 'invalid_attr', True):
             print 'should not print'
     # Test invalid value error
     with pytest.raises(ctx.NukeContextError):
-        with ctx.set_attr(node, 'file_type', 2.0, verbose=True):
+        with ctx.set_attr(node, 'file_type', 2.0):
             print 'should not print'
 
 
@@ -41,17 +41,26 @@ def test_multiple(node):
     node['disable'].setValue(True)
     node['file_type'].setValue('exr')
 
-    assert node['disable'].value() == True
+    assert node['disable'].value()
     assert node['file_type'].value() == 'exr'
     assert node['channels'].value() == 'rgb'
 
     with ctx.multiple_contexts([ctx1, ctx2, ctx3]):
-        assert node['disable'].value() == False
+        assert not node['disable'].value()
         assert node['file_type'].value() == 'jpeg'
         assert node['channels'].value() == 'rgba'
 
-    assert node['disable'].value() == True
+    assert node['disable'].value()
     assert node['file_type'].value() == 'exr'
     assert node['channels'].value() == 'rgb'
 
     node['disable'].setValue(False)
+
+
+def test_progress(node):
+    ctx1 = ctx.enabled(node)
+    iterable = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
+
+    with ctx1, ctx.Progress(iterable) as progress:
+        for idx, item in enumerate(progress):
+            assert item == idx + 1