def test_normal_same_table(self, con_a0, con_a1):
out_db_path = "test.sqlite"
runner = CliRunner()
with runner.isolated_filesystem():
result = runner.invoke(
cmd,
[
"file", con_a0.database_path, con_a1.database_path,
"-o", out_db_path,
])
print_traceback(result)
assert result.exit_code == ExitCode.SUCCESS
expected = TableData(
table_name=TEST_TABLE_NAME_A,
header_list=["attr_a", "attr_b"],
record_list=[
[1, 2],
[3, 4],
[11, 12],
[13, 14],
])
for tabledata in SqliteFileLoader(out_db_path).load():
assert tabledata == expected
def test_normal_multi_table(self, con_a0, con_b0):
out_db_path = "test.sqlite"
runner = CliRunner()
with runner.isolated_filesystem():
result = runner.invoke(
cmd,
[
"file", con_a0.database_path, con_b0.database_path,
"-o", out_db_path,
])
print_traceback(result)
assert result.exit_code == ExitCode.SUCCESS
expected_list = [
TableData(
table_name=TEST_TABLE_NAME_A,
header_list=["attr_a", "attr_b"],
record_list=[
[1, 2],
[3, 4],
]),
TableData(
table_name=TEST_TABLE_NAME_B,
header_list=["ba", "bb"],
record_list=[
[101, 102],
[103, 104],
]),
]
for tabledata in SqliteFileLoader(out_db_path).load():
print("[actual] {}".format(tabledata))
assert tabledata in expected_list
def test_normal_multi_table(self, con_a0, con_b0):
out_db_path = "test.sqlite"
runner = CliRunner()
with runner.isolated_filesystem():
result = runner.invoke(cmd, [
"-o", out_db_path, "file", con_a0.database_path,
con_b0.database_path
])
print_traceback(result)
assert result.exit_code == ExitCode.SUCCESS
expected_list = [
TableData(TEST_TABLE_NAME_A, ["attr_a", "attr_b"],
[[1, 2], [3, 4]]),
TableData(TEST_TABLE_NAME_B, ["ba", "bb"],
[[101, 102], [103, 104]]),
]
for tabledata in SqliteFileLoader(out_db_path).load():
if tabledata.table_name == SourceInfo.get_table_name():
continue
print("[actual]\n{}".format(tabledata))
for record in tabledata.value_matrix:
print(" {}".format(record))
assert tabledata in expected_list
def test_normal_multi_table(self, tmpdir):
test_filepath = str(tmpdir.join("test.sqlite"))
data_list = [
TableData(
table_name="first",
header_list=["ha1", "hb1", "hc1"],
row_list=[[1.0, 2.0, 3.0], [11.0, 12.0, 13.0]],
),
TableData(
table_name="second",
header_list=["ha2", "hb2", "hc2"],
row_list=[[11.0, 12.0, 13.0], [1.0, 2.0, 3.0]],
),
]
writer = ptw.SqliteTableWriter()
for data in data_list:
writer.from_tabledata(data)
writer.dump(test_filepath, close_after_write=False)
writer.close()
count = 0
for data, expected in zip(data_list,
SqliteFileLoader(test_filepath).load()):
assert data == expected
count += 1
assert count == 2
def test_normal_single_table(self, tmpdir):
test_filepath = str(tmpdir.join("test.sqlite"))
data = TableData(
"tablename", ["ha", "hb", "hc"], [[1.0, 2.0, 3.0], [11.0, 12.0, 13.0], [1.0, 2.0, 3.0]]
)
writer = ptw.SqliteTableWriter()
writer.from_tabledata(data)
writer.dump(test_filepath)
for expected in SqliteFileLoader(test_filepath).load():
assert data == expected
def test_normal(self, tmpdir, table, header, value, expected):
test_file_path = tmpdir.join("test.sqlite")
writer = ptw.SqliteTableWriter()
writer.open(str(test_file_path))
writer.table_name = table
writer.headers = header
writer.value_matrix = value
writer.iteration_length = len(value)
writer.write_table_iter()
writer.close()
for table_data in SqliteFileLoader(str(test_file_path)).load():
assert table_data == expected
def test_normal(self, tmpdir, table, header, value, expected):
test_file_path = tmpdir.join("test.sqlite")
writer = ptw.SqliteTableWriter()
writer.open(str(test_file_path))
writer.table_name = table
writer.headers = header
writer.value_matrix = value
writer.write_table()
writer.close()
for table_data in SqliteFileLoader(str(test_file_path)).load():
expected_dump = ptw.dump_tabledata(expected)
actual_dump = ptw.dump_tabledata(table_data)
print_test_result(expected=expected_dump, actual=actual_dump)
assert actual_dump == expected_dump
