blob: 170b83b449c28517a0b0c52aa808619f004bbfe1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
# gemato: Test utility functions
# vim:fileencoding=utf-8
# (c) 2017 Michał Górny
# Licensed under the terms of 2-clause BSD license
import io
import os
import os.path
import tempfile
import unittest
class TempDirTestCase(unittest.TestCase):
DIRS = []
FILES = {}
def setUp(self):
self.dir = tempfile.mkdtemp()
for k in self.DIRS:
os.mkdir(os.path.join(self.dir, k))
for k, v in self.FILES.items():
with io.open(os.path.join(self.dir, k), 'w', encoding='utf8') as f:
f.write(v)
def tearDown(self):
for k in self.FILES:
os.unlink(os.path.join(self.dir, k))
for k in reversed(self.DIRS):
os.rmdir(os.path.join(self.dir, k))
os.rmdir(self.dir)
|
