textual/tests/test_parser.py at main · PythonWorkbench/textual · GitHub

History

53 lines (42 loc) · 1.74 KB

Raw

from textual._parser import Parser

def test_read1():

class TestParser(Parser[str]):

"""A simple parser that reads a byte at a time from a stream."""

def parse(self, on_token):

while True:

data = yield self.read1()

if not data:

break

on_token(data)

test_parser = TestParser()

test_data = "Where there is a Will there is a way!"

for size in range(1, len(test_data) + 1):

# Feed the parser in pieces, first 1 character at a time, then 2, etc

data = []

for offset in range(0, len(test_data), size):

for chunk in test_parser.feed(test_data[offset : offset + size]):

data.append(chunk)

# Check we have received all the data in characters, no matter the fee dsize

assert len(data) == len(test_data)

assert "".join(data) == test_data

def test_read():

class TestParser(Parser[str]):

"""A parser that reads chunks of a given size from the stream."""

def __init__(self, size):

self.size = size

super().__init__()

def parse(self, on_token):

while True:

data = yield self.read1()

if not data:

break

on_token(data)

test_data = "Where there is a Will there is a way!"

for read_size in range(1, len(test_data) + 1):

for size in range(1, len(test_data) + 1):

test_parser = TestParser(read_size)

data = []

for offset in range(0, len(test_data), size):

for chunk in test_parser.feed(test_data[offset : offset + size]):

data.append(chunk)

assert "".join(data) == test_data

textual/tests/test_parser.py at main · PythonWorkbench/textual · GitHub

Uh oh!

Expand file tree

Breadcrumbs

test_parser.py

Latest commit

History

Breadcrumbs

test_parser.py

File metadata and controls