def test_fw():
# Run all the tests in a loop to have better confidance of randomisation.
for i in range(2, 5):
ncols = random.randint(1, 10)
default_specs = [(
random.randint(min_width, 10),
random.choice(type_choices),
) for i in range(ncols)]
colspecs = default_specs + [(10, 'date', '%d/%m/%Y', 2),
(10, 'float', 3),
(15, 'email', 'mail.com'),
(10, 'one_of', ['returnasis'])]
data = generate(colspecs, nrows=1)
expected_length = sum([c[0] for c in colspecs])
for d in data:
decoded = d.decode()
assert len(decoded) == expected_length
assert 'returnasis' in decoded
assert 'mail.com' in decoded
def test_generate_good_decode():
ncols = random.randint(1, 10)
colspecs = [(
random.randint(min_width, 10),
random.choice(type_choices),
) for i in range(ncols)]
good_decodes = [
('windows-1252', 'windows-1252'),
('utf-8', 'utf-8'),
('ascii', 'utf-8'),
]
for encoding, decoding in good_decodes:
for d in generate(colspecs + [(8, 'str')], nrows=1, encoding=encoding):
try:
print(d.decode(decoding))
assert True
except Exception as e:
assert False
def test_generate_bad_decode():
# has to have atleast 1 special char for this test to work
ncols = random.randint(1, 10)
colspecs = [(
random.randint(min_width, 10),
random.choice(type_choices),
) for i in range(ncols)]
# run the test more than once
bad_decodes = [('windows-1252', 'utf-8'), ('utf-8', 'ascii'),
('utf-16', 'utf-8')]
for encoding, decoding in bad_decodes:
for i in range(1, 3):
for d in generate(colspecs + [(8, 'str')],
nrows=1,
encoding=encoding):
try:
print(d.decode(decoding))
assert False
except UnicodeDecodeError as e:
assert True
def test_bad_date_length_throws_exception():
with pytest.raises(Exception) as e_info:
colspecs = [(8, 'date')]
for d in generate(colspecs, nrows=1):
assert d is not None
def test_bad_one_of_throws_exception_bad_length():
with pytest.raises(Exception) as e_info:
colspecs = [(5, 'one_of', ['returnasis'])]
for d in generate(colspecs, nrows=1):
assert False
def test_bad_float_length_throws_exception():
with pytest.raises(Exception) as e_info:
colspecs = [(4, 'float', 3)]
for d in generate(colspecs, nrows=1):
assert d is not None