Esempio n. 1
0
	def test_contains(self):
		ci = ConsumableIter([1, 2, 3, 4, 5])
		assert 1 in ci
		assert 5 in ci

		ci.consume()
		assert 1 not in ci # already consumed
		assert 5 in ci
Esempio n. 2
0
	def test_constructor(self):
		for v in (None, []):
			ci = ConsumableIter(v)
			assert ci.length == 0
			assert ci.remaining == 0

		ci = ConsumableIter([1, 2, 3])
		assert ci.length == 3
		assert ci.remaining == 3
Esempio n. 3
0
	def test_find_group_end(self):
		p = Parser()

		for items, group_end_pos in (
			(['foo', '==', '1', ')'], 3),
			(['foo', '==', '1', ')', 'or', 'a', '==', '1'], 3),
			(['foo', '==', '1', ')', 'or', '(', 'foo', '==', '1', ')'], 3),
		):
			assert p.find_group_end(ConsumableIter(items)) == group_end_pos

		with pytest.raises(Exception):
			assert p.find_group_end(ConsumableIter(['foo', '==', 'bar'])) == group_end_pos

		with pytest.raises(Exception) as e:
			assert p.find_group_end(ConsumableIter(['(', 'foo', '==', 'bar'])) == group_end_pos
Esempio n. 4
0
	def test_slice(self):
		ci = ConsumableIter([1, 2, 3])
		part = ci[1:2]
		assert isinstance(part, ConsumableIter)
		assert part.iterable == [2]
		assert part.length == 1
		assert part.remaining == 1
		assert part.pos == 0
Esempio n. 5
0
	def test_parse(self):
		for query, left, right, clazz in (
			('foo == bar', 'foo', 'bar', EqualsCondition),
			('foo > bar', 'foo', 'bar', GreaterThanCondition),
			('foo < bar', 'foo', 'bar', LessThanCondition),
			('foo >= bar', 'foo', 'bar', GreaterThanOrEqualCondition),
			('foo <= bar', 'foo', 'bar', LessThanOrEqualCondition),
		):
			q = Parser().parse(query).query
			assert q == ConsumableIter([
				clazz(left, right)
			])
			print("------")
Esempio n. 6
0
	def test_parse_and_or(self):
		for query, result in (
			("a == 1 and b == 2 or c == 3", # a and (b or c)
				AndOperator(
					EqualsCondition('a', '1'),
					OrOperator(
						EqualsCondition('b', '2'),
						EqualsCondition('c', '3'),
					)
				)
			),
			("a == 1 or b == 2 and c == 3",  # (a or b) and c
				AndOperator(
					OrOperator(
						EqualsCondition('a', '1'),
						EqualsCondition('b', '2')
					),
					EqualsCondition('c', '3')
				)
			),
			("a == 1 or b == 2 or c == 3",
				OrOperator(
					OrOperator(
						EqualsCondition('a', '1'),
						EqualsCondition('b', '2')
					),
					EqualsCondition('c', '3')
				)
			),
			("a == 1 and b == 2 and c == 3",
				AndOperator(
					AndOperator(
						EqualsCondition('a', '1'),
						EqualsCondition('b', '2')
					),
					EqualsCondition('c', '3')
				)
			),
		):
			q = Parser().parse(query).query
			assert q == ConsumableIter([result])
Esempio n. 7
0
	def test_consume_empty(self):
		ci = ConsumableIter([])

		with pytest.raises(ValueError):
			ci.consume(1)
Esempio n. 8
0
	def test_consume(self):
		ci = ConsumableIter([1, 2, 3])
		with pytest.raises(ValueError):
			ci.consume(None)

		ci = ConsumableIter([1, 2, 3])
		with pytest.raises(ValueError):
			ci.consume(-1)

		ci = ConsumableIter([1, 2, 3])
		with pytest.raises(ValueError):
			ci.consume(100)

		ci = ConsumableIter([1, 2, 3])
		ci.consume(1)
		with pytest.raises(ValueError):
			ci.consume(3)

		ci = ConsumableIter([1, 2, 3])
		assert ci.consume(0) is None
		assert ci.consume(1) == 1
		assert ci.consume(1) == 2
		assert ci.consume(1) == 3
		assert ci.remaining == 0
		assert not ci.has_more

		ci = ConsumableIter([1, 2, 3])
		assert ci.consume(3) == [1, 2, 3]
		assert ci.remaining == 0
		assert not ci.has_more
Esempio n. 9
0
	def test_has_more_empty(self):
		ci = ConsumableIter([])
		assert not ci.has_more
Esempio n. 10
0
	def test_has_more(self):
		ci = ConsumableIter([1, 2, 3])
		assert ci.has_more
Esempio n. 11
0
	def test_current_and_next_beyond_end(self):
		ci = ConsumableIter([1])
		ci.consume()
		assert ci.current is None
		assert ci.next is None
Esempio n. 12
0
	def test_current_and_next_empty(self):
		ci = ConsumableIter([])
		assert ci.current is None
		assert ci.next is None
Esempio n. 13
0
	def test_current_and_next(self):
		ci = ConsumableIter([1, 2, 3])
		assert ci.current == 1
		assert ci.next == 2
Esempio n. 14
0
	def test_iter(self):
		ci = ConsumableIter([1, 2, 3])

		assert [x for x in ci] == [1, 2, 3]
Esempio n. 15
0
	def test_len(self):
		ci = ConsumableIter([1, 2, 3])
		assert ci.length == 3
		assert len(ci) == 3