def testBraceExpand(self):
w = _assertReadWord(self, 'hi')
results = braces._BraceExpand(w.parts)
self.assertEqual(1, len(results))
for parts in results:
_PrettyPrint(ast.CompoundWord(parts))
print('')
w = _assertReadWord(self, 'B-{a,b}-E')
tree = braces._BraceDetect(w)
self.assertEqual(3, len(tree.parts))
pprint(tree)
results = braces._BraceExpand(tree.parts)
self.assertEqual(2, len(results))
for parts in results:
_PrettyPrint(ast.CompoundWord(parts))
print('')
w = _assertReadWord(self, 'B-{a,={b,c,d}=,e}-E')
tree = braces._BraceDetect(w)
self.assertEqual(3, len(tree.parts))
pprint(tree)
results = braces._BraceExpand(tree.parts)
self.assertEqual(5, len(results))
for parts in results:
_PrettyPrint(ast.CompoundWord(parts))
print('')
w = _assertReadWord(self, 'B-{a,b}-{c,d}-E')
tree = braces._BraceDetect(w)
self.assertEqual(5, len(tree.parts))
pprint(tree)
results = braces._BraceExpand(tree.parts)
self.assertEqual(4, len(results))
for parts in results:
_PrettyPrint(ast.CompoundWord(parts))
print('')
def testBraceDetect(self):
w = _assertReadWord(self, '}')
tree = braces._BraceDetect(w)
self.assertEqual(None, tree)
w = _assertReadWord(self, ',')
tree = braces._BraceDetect(w)
self.assertEqual(None, tree)
w = _assertReadWord(self, 'B-{a,b}-E')
tree = braces._BraceDetect(w)
self.assertEqual(3, len(tree.parts))
pprint(tree)
print('--')
# Multiple parts for each alternative
w = _assertReadWord(self, 'B-{a"a",b"b",c"c"}-E')
tree = braces._BraceDetect(w)
self.assertEqual(3, len(tree.parts))
pprint(tree)
print('--')
# Multiple expansion
w = _assertReadWord(self, 'B-{a,b}--{c,d}-E')
tree = braces._BraceDetect(w)
self.assertEqual(5, len(tree.parts))
pprint(tree)
print('--')
# Nested expansion
w = _assertReadWord(self, 'B-{a,b,c,={d,e}}-E')
tree = braces._BraceDetect(w)
pprint(tree)
self.assertEqual(3, len(tree.parts)) # B- {} -E
middle_part = tree.parts[1]
self.assertEqual(word_part_e.BracedAltPart, middle_part.tag)
self.assertEqual(4, len(middle_part.words)) # a b c ={d,e}
last_alternative = middle_part.words[3]
self.assertEqual(2, len(last_alternative.parts)) # = {d,e}
second_part = last_alternative.parts[1]
self.assertEqual(word_part_e.BracedAltPart, second_part.tag)
self.assertEqual(2, len(second_part.words)) # {d,e}
# Another nested expansion
w = _assertReadWord(self, 'B-{a,={b,c}=,d}-E')
tree = braces._BraceDetect(w)
pprint(tree)
self.assertEqual(3, len(tree.parts)) # B- {} -E
middle_part = tree.parts[1]
self.assertEqual(word_part_e.BracedAltPart, middle_part.tag)
self.assertEqual(3, len(middle_part.words)) # a ={b,c}= d
first_alternative = middle_part.words[0]
_PrettyPrint(first_alternative)
self.assertEqual(1, len(first_alternative.parts)) # a
#print('!!', first_alternative)
middle_alternative = middle_part.words[1]
self.assertEqual(3, len(middle_alternative.parts)) # = {b,c} =
middle_part2 = middle_alternative.parts[1]
self.assertEqual(word_part_e.BracedAltPart, middle_part2.tag)
self.assertEqual(2, len(middle_part2.words)) # b c
# Third alternative is a CompoundWord with zero parts
w = _assertReadWord(self, '{a,b,}')
tree = braces._BraceDetect(w)
pprint(tree)
self.assertEqual(1, len(tree.parts))
self.assertEqual(3, len(tree.parts[0].words))