
import tempfile
from unittest.runner import TextTestResult

from tap.directive import Directive
from tap.line import Result


class Factory(object):
    """A factory to produce commonly needed objects"""

    def make_ok(self, directive_text=''):
        return Result(
            True, 1, 'This is a description.', Directive(directive_text))

    def make_test_result(self):
        stream = tempfile.TemporaryFile()
        return TextTestResult(stream, None, 1)


import tempfile
try:
    from unittest.runner import TextTestResult
except ImportError:
    # Support Python 2.6.
    from unittest import _TextTestResult as TextTestResult

from tap.directive import Directive
from tap.line import Result


class Factory(object):
    """A factory to produce commonly needed objects"""

    def make_ok(self, directive_text=''):
        return Result(
            True, 1, 'This is a description.', Directive(directive_text))

    def make_test_result(self):
        stream = tempfile.TemporaryFile(mode='w')
        return TextTestResult(stream, None, 1)

Fix Python 2.6 and Python 3 tests.
Fix Python 2.6 and Python 3 tests.

  
  import tempfile
+ try:
- from unittest.runner import TextTestResult
+     from unittest.runner import TextTestResult
+ except ImportError:
+     # Support Python 2.6.
+     from unittest import _TextTestResult as TextTestResult
  
  from tap.directive import Directive
  from tap.line import Result
  
  
  class Factory(object):
      """A factory to produce commonly needed objects"""
  
      def make_ok(self, directive_text=''):
          return Result(
              True, 1, 'This is a description.', Directive(directive_text))
  
      def make_test_result(self):
-         stream = tempfile.TemporaryFile()
+         stream = tempfile.TemporaryFile(mode='w')
          return TextTestResult(stream, None, 1)
  
Fix Python 2.6 and Python 3 tests.
## Code Before:

import tempfile
from unittest.runner import TextTestResult

from tap.directive import Directive
from tap.line import Result


class Factory(object):
    """A factory to produce commonly needed objects"""

    def make_ok(self, directive_text=''):
        return Result(
            True, 1, 'This is a description.', Directive(directive_text))

    def make_test_result(self):
        stream = tempfile.TemporaryFile()
        return TextTestResult(stream, None, 1)

## Instruction:
Fix Python 2.6 and Python 3 tests.
## Code After:

import tempfile
try:
    from unittest.runner import TextTestResult
except ImportError:
    # Support Python 2.6.
    from unittest import _TextTestResult as TextTestResult

from tap.directive import Directive
from tap.line import Result


class Factory(object):
    """A factory to produce commonly needed objects"""

    def make_ok(self, directive_text=''):
        return Result(
            True, 1, 'This is a description.', Directive(directive_text))

    def make_test_result(self):
        stream = tempfile.TemporaryFile(mode='w')
        return TextTestResult(stream, None, 1)
