Fred Drake | 9175114 | 2001-03-21 18:29:25 +0000 | [diff] [blame] | 1 | """Test cases for the fnmatch module.""" |
| 2 | |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 3 | from test import support |
Fred Drake | 9175114 | 2001-03-21 18:29:25 +0000 | [diff] [blame] | 4 | import unittest |
| 5 | |
| 6 | from fnmatch import fnmatch, fnmatchcase |
| 7 | |
| 8 | |
| 9 | class FnmatchTestCase(unittest.TestCase): |
| 10 | def check_match(self, filename, pattern, should_match=1): |
| 11 | if should_match: |
Georg Brandl | ab91fde | 2009-08-13 08:51:18 +0000 | [diff] [blame] | 12 | self.assertTrue(fnmatch(filename, pattern), |
Fred Drake | 9175114 | 2001-03-21 18:29:25 +0000 | [diff] [blame] | 13 | "expected %r to match pattern %r" |
| 14 | % (filename, pattern)) |
| 15 | else: |
Georg Brandl | ab91fde | 2009-08-13 08:51:18 +0000 | [diff] [blame] | 16 | self.assertTrue(not fnmatch(filename, pattern), |
Fred Drake | 9175114 | 2001-03-21 18:29:25 +0000 | [diff] [blame] | 17 | "expected %r not to match pattern %r" |
| 18 | % (filename, pattern)) |
| 19 | |
| 20 | def test_fnmatch(self): |
| 21 | check = self.check_match |
| 22 | check('abc', 'abc') |
| 23 | check('abc', '?*?') |
| 24 | check('abc', '???*') |
| 25 | check('abc', '*???') |
| 26 | check('abc', '???') |
| 27 | check('abc', '*') |
| 28 | check('abc', 'ab[cd]') |
| 29 | check('abc', 'ab[!de]') |
| 30 | check('abc', 'ab[de]', 0) |
| 31 | check('a', '??', 0) |
| 32 | check('a', 'b', 0) |
| 33 | |
| 34 | # these test that '\' is handled correctly in character sets; |
Gregory P. Smith | 6c4a725 | 2009-11-01 20:36:24 +0000 | [diff] [blame^] | 35 | # see SF bug #409651 |
Fred Drake | 9175114 | 2001-03-21 18:29:25 +0000 | [diff] [blame] | 36 | check('\\', r'[\]') |
| 37 | check('a', r'[!\]') |
| 38 | check('\\', r'[!\]', 0) |
| 39 | |
Gregory P. Smith | 6c4a725 | 2009-11-01 20:36:24 +0000 | [diff] [blame^] | 40 | # test that filenames with newlines in them are handled correctly. |
| 41 | # http://bugs.python.org/issue6665 |
| 42 | check('foo\nbar', 'foo*') |
| 43 | check('foo\nbar\n', 'foo*') |
| 44 | check('\nfoo', 'foo*', False) |
| 45 | check('\n', '*') |
| 46 | |
Guido van Rossum | f0af3e3 | 2008-10-02 18:55:37 +0000 | [diff] [blame] | 47 | def test_mix_bytes_str(self): |
| 48 | self.assertRaises(TypeError, fnmatch, 'test', b'*') |
| 49 | self.assertRaises(TypeError, fnmatch, b'test', '*') |
| 50 | self.assertRaises(TypeError, fnmatchcase, 'test', b'*') |
| 51 | self.assertRaises(TypeError, fnmatchcase, b'test', '*') |
| 52 | |
| 53 | def test_bytes(self): |
| 54 | self.check_match(b'test', b'te*') |
| 55 | self.check_match(b'test\xff', b'te*\xff') |
Gregory P. Smith | 6c4a725 | 2009-11-01 20:36:24 +0000 | [diff] [blame^] | 56 | self.check_match(b'foo\nbar', b'foo*') |
| 57 | |
Fred Drake | 9175114 | 2001-03-21 18:29:25 +0000 | [diff] [blame] | 58 | |
Fred Drake | 2e2be37 | 2001-09-20 21:33:42 +0000 | [diff] [blame] | 59 | def test_main(): |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 60 | support.run_unittest(FnmatchTestCase) |
Fred Drake | 2e2be37 | 2001-09-20 21:33:42 +0000 | [diff] [blame] | 61 | |
| 62 | |
| 63 | if __name__ == "__main__": |
| 64 | test_main() |