Fix some warnings in tests

Browse Source
This commit is contained in:
Stefan Wehrmeyer 2015-01-06 21:57:32 +01:00
parent d2f36d6413
commit ee12dd45cc
2 changed files with 24 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
test/test_freeze.py
View File

@ -29,9 +29,9 @@ class FreezeTestCase(unittest.TestCase):
from dataset.freeze.app import freeze
freeze(self.db['weather'].all(), format='csv', filename='wäther.csv'.encode('utf8'), prefix=self.d)
self.assert_(os.path.exists(os.path.join(self.d, 'wäther.csv')))
self.assertTrue(os.path.exists(os.path.join(self.d, 'wäther.csv')))
freeze(self.db['weather'].all(), format='csv', filename='wäther.csv', prefix=self.d)
self.assert_(os.path.exists(os.path.join(self.d, 'wäther.csv')))
self.assertTrue(os.path.exists(os.path.join(self.d, 'wäther.csv')))
def test_freeze_csv(self):
from dataset.freeze.app import freeze
@ -43,6 +43,7 @@ class FreezeTestCase(unittest.TestCase):
fh = open(path, 'rt', encoding='utf8', newline='')
else:
fh = open(path, 'rU')
try:
rows = list(reader(fh))
keys = rows[0]
if not PY3:
@ -60,6 +61,8 @@ class FreezeTestCase(unittest.TestCase):
else:
v1 = '%s' % v1
self.assertEqual(v2, v1)
finally:
fh.close()
class SerializerTestCase(unittest.TestCase):
@ -74,4 +77,4 @@ class SerializerTestCase(unittest.TestCase):
s = Serializer(Export({'filename': 'f'}, {}), '')
s.wrap
s = Serializer(Export({'filename': '-'}, {}), '')
self.assert_(s.fileobj)
self.assertTrue(s.fileobj)

2
test/test_freeze_config.py
View File

@ -23,7 +23,7 @@ class TestConfiguration(unittest.TestCase):
self.assertTrue(exports[0].get_bool('bool'))
self.assertEqual(exports[0].get_int('nan', 'default'), 'default')
self.assertEqual(exports[0].get_int('number'), 5)
self.assert_(exports[0].name)
self.assertTrue(exports[0].name)
def test_exports_fail(self):
from dataset.freeze.config import Configuration