def test_cli_tree(self, output):
words = ["test"]
sys.argv = ["ety.py", "-t"] + words
ety.cli()
tree = ety.tree("test")
expected_lines = len(tree)
self.assertEqual(expected_lines, output.lines)
def test_cli_multiple_words_tree(self, output):
words = ["test", "word"]
sys.argv = ["ety.py", "-t"] + words
ety.cli()
trees = [ety.tree(word) for word in words]
expected_length = sum(len(tree) for tree in trees) + len(words) - 1
self.assertEqual(expected_length, output.lines)
def test_cli_multiple_words(self, output):
words = ["test", "word"]
sys.argv = ["ety.py"] + words
ety.cli()
origins = [origin for word in words for origin in ety.origins(word)]
expected_lines = len(words) + len(origins) + len(words) - 1
self.assertEqual(expected_lines, output.lines)
def test_cli_recursive(self, output):
words = ["test"]
sys.argv = ["ety.py", "-r"] + words
ety.cli()
origins = ety.origins("test", recursive=True)
expected_lines = len(words) + len(origins)
self.assertEqual(expected_lines, output.lines)
def test_cli_no_args(self, output):
words = ["test"]
sys.argv = ["ety.py", "test"]
ety.cli()
origins = ety.origins("test")
expected_lines = len(words) + len(origins)
self.assertEqual(expected_lines, output.lines)
#!/usr/local/bin/python
# -*- coding: utf-8 -*-
from ety import cli
if __name__ == '__main__':
cli()
