16
16
import warnings
17
17
from collections import OrderedDict , defaultdict
18
18
19
- # for python 3 compatibility
20
- import six
21
-
22
- try :
23
- from Queue import Empty
24
- except :
25
- from queue import Empty
19
+ from queue import Empty
26
20
27
21
# for reading notebook files
28
22
import nbformat
@@ -503,9 +497,9 @@ def compare_outputs(self, test, ref, skip_compare=None):
503
497
504
498
def format_output_compare (self , key , left , right ):
505
499
"""Format an output for printing"""
506
- if isinstance (left , six . string_types ):
500
+ if isinstance (left , str ):
507
501
left = _trim_base64 (left )
508
- if isinstance (right , six . string_types ):
502
+ if isinstance (right , str ):
509
503
right = _trim_base64 (right )
510
504
511
505
cc = self .colors
@@ -799,7 +793,7 @@ def sanitize_outputs(self, outputs, skip_sanitize=('metadata',
799
793
def sanitize (self , s ):
800
794
"""sanitize a string for comparison.
801
795
"""
802
- if not isinstance (s , six . string_types ):
796
+ if not isinstance (s , str ):
803
797
return s
804
798
805
799
"""
@@ -808,7 +802,7 @@ def sanitize(self, s):
808
802
is passed when py.test is called. Otherwise, the strings
809
803
are not processed
810
804
"""
811
- for regex , replace in six . iteritems ( self .parent .sanitize_patterns ):
805
+ for regex , replace in self .parent .sanitize_patterns . items ( ):
812
806
s = re .sub (regex , replace , s )
813
807
return s
814
808
@@ -904,6 +898,6 @@ def _trim_base64(s):
904
898
905
899
def _indent (s , indent = ' ' ):
906
900
"""Intent each line with indent"""
907
- if isinstance (s , six . string_types ):
901
+ if isinstance (s , str ):
908
902
return '\n ' .join (('%s%s' % (indent , line ) for line in s .splitlines ()))
909
903
return s
0 commit comments