def test_file_record_number_with_read_limit(self):
self.MyTest = mod.FileTyper(self.test1_fqfn, read_limit=10)
self.MyTest.analyze_file()
(est_rec_cnt, est_flag) = self.MyTest._count_records()
assert est_flag is True
# it should normally be within 5-10% of 100
assert abs(115 - est_rec_cnt) < 30
def test_file_record_number_with_read_limit(self):
file_typer = mod.FileTyper(self.dialect, self.test_fqfn, read_limit=10)
file_typer.analyze_file()
(est_rec_cnt, est_flag) = file_typer._count_records()
assert est_flag is True
# it should normally be within 5-10% of 100
assert abs(115 - est_rec_cnt) < 30
def setup_method(self, method):
self.record_cnt = 100
self.delimiter = '|'
self.quoting = False
self.test1_fqfn = generate_test_file1(self.delimiter, self.quoting,
self.record_cnt)
self.MyTest = mod.FileTyper(self.test1_fqfn, None, None, None)
self.MyTest.analyze_file()
def setup_method(self, method):
self.record_cnt = 100
self.dialect = csvhelper.Dialect(delimiter='|',
quoting=csv.QUOTE_NONE,
has_header=False)
self.test_fqfn = generate_test_file1(self.dialect, self.record_cnt)
self.file_typer = mod.FileTyper(self.dialect, self.test_fqfn)
self.file_typer.analyze_file()
def test_quote_all(self):
dialect = csvhelper.Dialect(delimiter='|',
quoting=csv.QUOTE_ALL,
has_header=False)
self.test_fqfn = generate_test_file1(dialect, self.record_cnt)
file_typer = mod.FileTyper(dialect, self.test_fqfn)
file_typer.analyze_file()
assert file_typer.record_cnt == self.record_cnt
assert file_typer.field_cnt == 4
def test_quote_all(self):
self.test1_fqfn = generate_test_file1(self.delimiter, csv.QUOTE_ALL,
self.record_cnt)
self.MyTest = mod.FileTyper(self.test1_fqfn)
self.MyTest.analyze_file()
assert self.MyTest.record_cnt == self.record_cnt
assert self.MyTest.field_cnt == 4
assert self.MyTest.format_type == 'csv'
assert self.MyTest.dialect.delimiter == self.delimiter
assert self.MyTest.dialect.quoting == csv.QUOTE_ALL