Fix to check dict status code response in test_container_options

base-notebook/test/test_container_options.py

@@ -68,7 +68,7 @@ def test_sudo(container):
         command=['start.sh', 'sudo', 'id']
     )
     rv = c.wait(timeout=10)
-    assert rv == 0
+    assert rv == 0 or rv["StatusCode"] == 0
     assert 'uid=0(root)' in c.logs(stdout=True).decode('utf-8')
 
 
@@ -81,7 +81,7 @@ def test_sudo_path(container):
         command=['start.sh', 'sudo', 'which', 'jupyter']
     )
     rv = c.wait(timeout=10)
-    assert rv == 0
+    assert rv == 0 or rv["StatusCode"] == 0
     assert c.logs(stdout=True).decode('utf-8').rstrip().endswith('/opt/conda/bin/jupyter')
 
 
@@ -93,7 +93,7 @@ def test_sudo_path_without_grant(container):
         command=['start.sh', 'which', 'jupyter']
     )
     rv = c.wait(timeout=10)
-    assert rv == 0
+    assert rv == 0 or rv["StatusCode"] == 0
     assert c.logs(stdout=True).decode('utf-8').rstrip().endswith('/opt/conda/bin/jupyter')
 
 
@@ -107,5 +107,5 @@ def test_group_add(container, tmpdir):
         command=['start.sh', 'id']
     )
     rv = c.wait(timeout=5)
-    assert rv == 0
+    assert rv == 0 or rv["StatusCode"] == 0
     assert 'uid=1010 gid=1010 groups=1010,100(users)' in c.logs(stdout=True).decode('utf-8')