class TestRemove(unittest.TestCase): def setUp(self): self.temporary_directory = tempfile.mkdtemp() self.temporary_file = tempfile.NamedTemporaryFile(delete=False).name def test_remove(self): for path in (self.temporary_file, self.temporary_directory): self.assertTrue(os.path.exists(path)) remove(path) self.assertFalse(os.path.exists(path)) class TestMakedirs(unittest.TestCase): def setUp(self): self.temporary_directory = tempfile.mkdtemp() def tearDown(self): shutil.rmtree(self.temporary_directory) def test_makedirs(self): path = os.path.join(self.temporary_directory, 'foo', 'bar', 'baz') self.assertFalse(os.path.exists(path)) makedirs(path) self.assertTrue(os.path.exists(path)) if __name__ == '__main__': unittest.main()
# pylint: disable=C0111 # pylint: disable=R0904 from __future__ import absolute_import from gutenberg._domain_model.exceptions import InvalidEtextIdException from gutenberg._domain_model.types import validate_etextno from tests._util import unittest class TestValidateEtextno(unittest.TestCase): def test_is_valid_etext(self): self.assertIsNotNone(validate_etextno(1)) self.assertIsNotNone(validate_etextno(12)) self.assertIsNotNone(validate_etextno(123)) self.assertIsNotNone(validate_etextno(1234)) def test_is_invalid_etext(self): self.assertRaises(InvalidEtextIdException, validate_etextno, 'not-int') self.assertRaises(InvalidEtextIdException, validate_etextno, -123) self.assertRaises(InvalidEtextIdException, validate_etextno, 0) self.assertRaises(InvalidEtextIdException, validate_etextno, 12.3) if __name__ == '__main__': unittest.main()