def test_get_temp_filename(self, settings):
        nb_temp_files_before = len(os.listdir(self.temp_dir))

        temp_filename = tmp.get_temp_filename()

        self.assertIn(self.temp_dir, temp_filename)

        os.remove(temp_filename)

        extension = '.png'

        temp_filename = tmp.get_temp_filename(extension)

        self.assertIn(extension, temp_filename)

        os.remove(temp_filename)

        self.assertEquals(len(os.listdir(self.temp_dir)), nb_temp_files_before)
    def test_get_temp_filename(self, settings):
        nb_temp_files_before = len(os.listdir(self.temp_dir))

        temp_filename = tmp.get_temp_filename()

        self.assertIn(self.temp_dir, temp_filename)

        os.remove(temp_filename)

        extension = '.png'

        temp_filename = tmp.get_temp_filename(extension)

        self.assertIn(extension, temp_filename)

        os.remove(temp_filename)

        self.assertEquals(len(os.listdir(self.temp_dir)), nb_temp_files_before)
Example #3
0
    def test(self):
        def _path_from_name(name):
            return os.path.join(os.path.dirname(__file__), 'results/' + name)

        def _compare(file1, file2, **kwargs):
            self.assertImageAlmostEqual(_path_from_name(file1), _path_from_name(file2), **kwargs)

        # Compare identical images
        _compare('white.png', 'white.png')
        _compare('red.jpg', 'red.jpg')
        # Compare close images
        _compare('gradient.png', 'gradient-broken.png', threshold=7, diff_file=tmp.get_temp_filename('.png'))

        with self.assertRaises(AssertionError):
            _compare('red.jpg', 'white.png')