5 import shutil

     5 import shutil

     6 import tempfile

     6 import tempfile

     7 import unittest

     7 import unittest

     8 

     8 

     9 from mercurial import (

     9 from mercurial import (

    10     util,

    11     util,

    11 )

    12 )

    12 atomictempfile = util.atomictempfile

    13 atomictempfile = util.atomictempfile

    13 

    17 

    14 class testatomictempfile(unittest.TestCase):

    18 class testatomictempfile(unittest.TestCase):

    15     def setUp(self):

    19     def setUp(self):

    16         self._testdir = tempfile.mkdtemp('atomictempfiletest')

    20         self._testdir = tempfile.mkdtemp('atomictempfiletest')

    17         self._filename = os.path.join(self._testdir, 'testfilename')

    21         self._filename = os.path.join(self._testdir, 'testfilename')