1616import warnings
1717from collections import OrderedDict , defaultdict
1818
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
2620
2721# for reading notebook files
2822import nbformat
@@ -503,9 +497,9 @@ def compare_outputs(self, test, ref, skip_compare=None):
503497
504498 def format_output_compare (self , key , left , right ):
505499 """Format an output for printing"""
506- if isinstance (left , six . string_types ):
500+ if isinstance (left , str ):
507501 left = _trim_base64 (left )
508- if isinstance (right , six . string_types ):
502+ if isinstance (right , str ):
509503 right = _trim_base64 (right )
510504
511505 cc = self .colors
@@ -799,7 +793,7 @@ def sanitize_outputs(self, outputs, skip_sanitize=('metadata',
799793 def sanitize (self , s ):
800794 """sanitize a string for comparison.
801795 """
802- if not isinstance (s , six . string_types ):
796+ if not isinstance (s , str ):
803797 return s
804798
805799 """
@@ -808,7 +802,7 @@ def sanitize(self, s):
808802 is passed when py.test is called. Otherwise, the strings
809803 are not processed
810804 """
811- for regex , replace in six . iteritems ( self .parent .sanitize_patterns ):
805+ for regex , replace in self .parent .sanitize_patterns . items ( ):
812806 s = re .sub (regex , replace , s )
813807 return s
814808
@@ -904,6 +898,6 @@ def _trim_base64(s):
904898
905899def _indent (s , indent = ' ' ):
906900 """Intent each line with indent"""
907- if isinstance (s , six . string_types ):
901+ if isinstance (s , str ):
908902 return '\n ' .join (('%s%s' % (indent , line ) for line in s .splitlines ()))
909903 return s
0 commit comments