mirror of
https://github.com/wassname/segpy.git
synced 2026-07-02 04:59:31 +08:00
Robert Smallshire
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
from collections import namedtuple, Counter
|
|
import random
|
|
import unittest
|
|
from hypothesis import given
|
|
from hypothesis.descriptors import one_of, SampledFrom, Just, sampled_from, just
|
|
from hypothesis.searchstrategy import MappedSearchStrategy, StringStrategy
|
|
from hypothesis.strategytable import StrategyTable
|
|
from segpy.encoding import EBCDIC, ASCII
|
|
from segpy.toolkit import format_extended_textual_header, CARDS_PER_HEADER, END_TEXT_STANZA, CARD_LENGTH
|
|
|
|
|
|
# class MultiLineString(str):
|
|
# pass
|
|
#
|
|
#
|
|
# class MultiLineStringStrategy(MappedSearchStrategy):
|
|
#
|
|
# def pack(self, x):
|
|
# return '\n'.join(x)
|
|
#
|
|
# def unpack(self, x):
|
|
# return ''.join(x.splitlines())
|
|
#
|
|
#
|
|
# StrategyTable.default().define_specification_for(
|
|
# MultiLineString,
|
|
# lambda s, d: MultiLineStringStrategy(
|
|
# strategy=s.strategy([str]),
|
|
# descriptor=MultiLineString))
|
|
|
|
|
|
class TestFormatExtendedTextualHeader(unittest.TestCase):
|
|
|
|
@given(str,
|
|
sampled_from([ASCII, EBCDIC]),
|
|
bool)
|
|
def test_forty_lines_per_page(self, text, encoding, include_text_stop):
|
|
pages = format_extended_textual_header(text, encoding, include_text_stop)
|
|
self.assertTrue(all(len(page) == CARDS_PER_HEADER for page in pages))
|
|
|
|
@given(str,
|
|
sampled_from([ASCII, EBCDIC]),
|
|
bool)
|
|
def test_eighty_bytes_per_encoded_line(self, text, encoding, include_text_stop):
|
|
pages = format_extended_textual_header(text, encoding, include_text_stop)
|
|
self.assertTrue(all([len(line.encode(encoding)) == CARD_LENGTH for page in pages for line in page]))
|
|
|
|
@given(str,
|
|
sampled_from([ASCII, EBCDIC]),
|
|
bool)
|
|
def test_lines_end_with_cr_lf(self, text, encoding, include_text_stop):
|
|
pages = format_extended_textual_header(text, encoding, include_text_stop)
|
|
self.assertTrue(all([line.endswith('\r\n') for page in pages for line in page]))
|
|
|
|
@given(str,
|
|
sampled_from([ASCII, EBCDIC]),
|
|
just(True))
|
|
def test_lines_end_with_cr_lf(self, text, encoding, include_text_stop):
|
|
pages = format_extended_textual_header(text, encoding, include_text_stop)
|
|
self.assertTrue(pages[-1][0].startswith(END_TEXT_STANZA))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |