def test_delims(self): args = _parse_args('-w --include-chars !abcdefghi'.split()) self.assertEqual(set(ch for ch in args.include_chars), set(ch for ch in "!abcdefghi")) self.assertEqual(set(ch for ch in args.exclude_chars), set()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) for ch in '!abcdefghi': self.assertTrue(ch in args.delimset) args = _parse_args(['--word', '--exclude-chars', "./, \n\ta"]) self.assertEqual(set(ch for ch in args.exclude_chars), set(ch for ch in "./, \n\ta")) self.assertEqual(set(ch for ch in args.include_chars), set()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) for ch in "./, \n\ta": self.assertFalse(ch in args.delimset) args = _parse_args( '--include-chars 012345678 --exclude-chars $%@!# -w 4'.split()) self.assertEqual(set(ch for ch in args.exclude_chars), set(ch for ch in "$%@!#")) self.assertEqual(set(ch for ch in args.include_chars), set(str(i) for i in range(9))) self.assertTrue(args.word) self.assertEqual(args.word_level, 4) for i in range(9): self.assertTrue(str(i) in args.delimset) for ch in "$%@!#": self.assertFalse(ch in args.delimset)
def test_delims(self): args = _parse_args('-w --include-chars !abcdefghi'.split()) self.assertEqual(set(ch for ch in args.include_chars), set(ch for ch in "!abcdefghi")) self.assertEqual(set(ch for ch in args.exclude_chars), set()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) for ch in '!abcdefghi': self.assertTrue(ch in args.delimset) args = _parse_args(['--word', '--exclude-chars', "./, \n\ta"]) self.assertEqual(set(ch for ch in args.exclude_chars), set(ch for ch in "./, \n\ta")) self.assertEqual(set(ch for ch in args.include_chars), set()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) for ch in "./, \n\ta": self.assertFalse(ch in args.delimset) args = _parse_args('--include-chars 012345678 --exclude-chars $%@!# -w 4'.split()) self.assertEqual(set(ch for ch in args.exclude_chars), set(ch for ch in "$%@!#")) self.assertEqual(set(ch for ch in args.include_chars), set(str(i) for i in range(9))) self.assertTrue(args.word) self.assertEqual(args.word_level, 4) for i in range(9): self.assertTrue(str(i) in args.delimset) for ch in "$%@!#": self.assertFalse(ch in args.delimset)
def test_wc(self): args = _parse_args('-w -c'.split()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.ignore_lines) args = _parse_args('-c --word'.split()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.ignore_lines) args = _parse_args('--word -c 10 --ignore-lines'.split()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertTrue(args.character) self.assertEqual(args.character_level, 10) self.assertTrue(args.ignore_lines) args = _parse_args('--ignore-lines -c 3 --word 10'.split()) self.assertTrue(args.word) self.assertEqual(args.word_level, 10) self.assertTrue(args.character) self.assertEqual(args.character_level, 3) self.assertTrue(args.ignore_lines)
def test_wc(self): args = _parse_args('-w -c'.split()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.ignore_lines) args = _parse_args('-c --word'.split()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.ignore_lines) args = _parse_args('--word -c 10 --ignore-lines'.split()) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertTrue(args.character) self.assertEqual(args.character_level, 10) self.assertTrue(args.ignore_lines) args = _parse_args('--ignore-lines -c 3 --word 10'.split()) self.assertTrue(args.word) self.assertEqual(args.word_level, 10) self.assertTrue(args.character) self.assertEqual(args.character_level, 3) self.assertTrue(args.ignore_lines)
def test_no_args(self): args = _parse_args([]) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.word) self.assertFalse(args.byte) self.assertIs(args.input, sys.stdin) self.assertIs(args.output, sys.stdout) self.assertEqual(args.include_chars, '') self.assertEqual(args.exclude_chars, '')
def test_no_args(self): args = _parse_args([]) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.word) self.assertFalse(args.byte) self.assertIs(args.input, sys.stdin) self.assertIs(args.output, sys.stdout) self.assertEqual(args.include_chars, '') self.assertEqual(args.exclude_chars, '')
def test_bytes(self): args = _parse_args(['-b']) self.assertTrue(args.byte) self.assertEqual(args.byte_level, 1) self.assertFalse(args.character) self.assertFalse(args.word) self.assertIs(args.input, sys.stdin.buffer) self.assertIs(args.output, sys.stdout.buffer) args = _parse_args(['--byte']) self.assertTrue(args.byte) self.assertEqual(args.byte_level, 1) self.assertFalse(args.character) self.assertFalse(args.word) args = _parse_args('--byte 4'.split()) self.assertTrue(args.byte) self.assertEqual(args.byte_level, 4) self.assertFalse(args.character) self.assertFalse(args.word)
def test_character_args(self): args = _parse_args(['-c']) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.word) self.assertFalse(args.byte) self.assertIs(args.input, sys.stdin) self.assertIs(args.output, sys.stdout) args = _parse_args(['--character']) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.word) self.assertFalse(args.byte) args = _parse_args(['-c 24']) self.assertTrue(args.character) self.assertEqual(args.character_level, 24) self.assertFalse(args.word) self.assertFalse(args.byte)
def test_bytes(self): args = _parse_args(['-b']) self.assertTrue(args.byte) self.assertEqual(args.byte_level, 1) self.assertFalse(args.character) self.assertFalse(args.word) self.assertIs(args.input, sys.stdin.buffer) self.assertIs(args.output, sys.stdout.buffer) args = _parse_args(['--byte']) self.assertTrue(args.byte) self.assertEqual(args.byte_level, 1) self.assertFalse(args.character) self.assertFalse(args.word) args = _parse_args('--byte 4'.split()) self.assertTrue(args.byte) self.assertEqual(args.byte_level, 4) self.assertFalse(args.character) self.assertFalse(args.word)
def test_character_args(self): args = _parse_args(['-c']) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.word) self.assertFalse(args.byte) self.assertIs(args.input, sys.stdin) self.assertIs(args.output, sys.stdout) args = _parse_args(['--character']) self.assertTrue(args.character) self.assertEqual(args.character_level, 1) self.assertFalse(args.word) self.assertFalse(args.byte) args = _parse_args(['-c 24']) self.assertTrue(args.character) self.assertEqual(args.character_level, 24) self.assertFalse(args.word) self.assertFalse(args.byte)
def test_word_args(self): args = _parse_args(['-w']) self.assertFalse(args.character) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertFalse(args.byte) self.assertFalse(args.ignore_lines) self.assertIs(args.input, sys.stdin) self.assertIs(args.output, sys.stdout) args = _parse_args(['--word']) self.assertFalse(args.character) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertFalse(args.byte) args = _parse_args('--word 3'.split()) self.assertFalse(args.character) self.assertTrue(args.word) self.assertEqual(args.word_level, 3) self.assertFalse(args.byte)
def test_word_args(self): args = _parse_args(['-w']) self.assertFalse(args.character) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertFalse(args.byte) self.assertFalse(args.ignore_lines) self.assertIs(args.input, sys.stdin) self.assertIs(args.output, sys.stdout) args = _parse_args(['--word']) self.assertFalse(args.character) self.assertTrue(args.word) self.assertEqual(args.word_level, 1) self.assertFalse(args.byte) args = _parse_args('--word 3'.split()) self.assertFalse(args.character) self.assertTrue(args.word) self.assertEqual(args.word_level, 3) self.assertFalse(args.byte)
def test_io(self): import tempfile input_file = tempfile.NamedTemporaryFile(delete=False) input_file.write(b'testing testing') input_file.close() args = _parse_args([input_file.name]) self.assertEqual(args.input.read(), 'testing testing') args.input.close() self.assertIs(args.output, sys.stdout) output_file = tempfile.NamedTemporaryFile(delete=False) args = _parse_args(['--output', output_file.name]) args.output.write('abc def') args.output.close() self.assertEqual(output_file.read(), b'abc def') output_file.close() self.assertIs(args.input, sys.stdin) input_file = tempfile.NamedTemporaryFile(delete=False) input_file.write(b'this') input_file.close() output_file = tempfile.NamedTemporaryFile(delete=False) args = _parse_args([input_file.name, '--output', output_file.name]) self.assertEqual(args.input.read(), 'this') args.input.close() args.output.write('that') args.output.close() self.assertEqual(output_file.read(), b'that') output_file.close() same_file = tempfile.NamedTemporaryFile(delete=False) args = _parse_args(['--output', same_file.name, same_file.name]) args.output.write('goodbye') args.output.close() self.assertEqual(same_file.read(), b'goodbye') same_file.write(b'hello') same_file.close() self.assertEqual(args.input.read(), 'goodbyehello') args.input.close()
def test_io(self): import tempfile input_file = tempfile.NamedTemporaryFile(delete=False) input_file.write(b'testing testing') input_file.close() args = _parse_args([input_file.name]) self.assertEqual(args.input.read(), 'testing testing') args.input.close() self.assertIs(args.output, sys.stdout) output_file = tempfile.NamedTemporaryFile(delete=False) args = _parse_args(['--output', output_file.name]) args.output.write('abc def') args.output.close() self.assertEqual(output_file.read(), b'abc def') output_file.close() self.assertIs(args.input, sys.stdin) input_file = tempfile.NamedTemporaryFile(delete=False) input_file.write(b'this') input_file.close() output_file = tempfile.NamedTemporaryFile(delete=False) args = _parse_args([input_file.name, '--output', output_file.name]) self.assertEqual(args.input.read(), 'this') args.input.close() args.output.write('that') args.output.close() self.assertEqual(output_file.read(), b'that') output_file.close() same_file = tempfile.NamedTemporaryFile(delete=False) args = _parse_args(['--output', same_file.name, same_file.name]) args.output.write('goodbye') args.output.close() self.assertEqual(same_file.read(), b'goodbye') same_file.write(b'hello') same_file.close() self.assertEqual(args.input.read(), 'goodbyehello') args.input.close()
def _parse_err(cls, arg_str): with cls._nostderr(): _parse_args(arg_str.split())
def _parse_err(cls, arg_str): with cls._nostderr(): _parse_args(arg_str.split())