def setUp(self):
app.config["TESTING"] = True
self.app = app.test_client()
self.content = OrderedDict()
self.content.update(
{"Sheet1": [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]]})
self.content.update(
{"Sheet2": [[4, 4, 4, 4], [5, 5, 5, 5], [6, 6, 6, 6]]})
self.content.update(
{"Sheet3": [[u"X", u"Y", u"Z"], [1, 4, 7], [2, 5, 8], [3, 6, 9]]})
def setUp(self):
self.testfile = "multiple1.csvz"
self.content = OrderedDict()
self.content.update(
{"Sheet1": [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]]})
self.content.update(
{"Sheet2": [[4, 4, 4, 4], [5, 5, 5, 5], [6, 6, 6, 6]]})
self.content.update(
{"Sheet3": [[u'X', u'Y', u'Z'], [1, 4, 7], [2, 5, 8], [3, 6, 9]]})
self._write_test_file(self.testfile, self.content)
def test_issue_10():
thedict = OrderedDict()
thedict.update({"Column 1": [1, 2, 3]})
thedict.update({"Column 2": [1, 2, 3]})
thedict.update({"Column 3": [1, 2, 3]})
pe.save_as(adict=thedict, dest_file_name="issue10.xls")
newdict = pe.get_dict(file_name="issue10.xls")
assert isinstance(newdict, OrderedDict) is True
assert thedict == newdict
os.unlink("issue10.xls")
def setUp(self):
self.testfile = "multiple1.xls"
self.content = OrderedDict()
self.content.update(
{"Sheet1": [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]]})
self.content.update(
{"Sheet2": [[4, 4, 4, 4], [5, 5, 5, 5], [6, 6, 6, 6]]})
self.content.update(
{"Sheet3": [[u'X', u'Y', u'Z'], [1, 4, 7], [2, 5, 8], [3, 6, 9]]})
pe.save_book_as(dest_file_name=self.testfile, bookdict=self.content)
def test_get_book_from_sql(self):
book_dict = pe.get_book_dict(
session=Session(), tables=[Signature, Signature2]
)
expected = OrderedDict()
expected.update({"signature": [["X", "Y", "Z"], [1, 2, 3], [4, 5, 6]]})
expected.update(
{"signature2": [["A", "B", "C"], [1, 2, 3], [4, 5, 6]]}
)
assert book_dict == expected
def test_save_book_to_memory_from_sql(self):
test_file = pe.save_book_as(dest_file_type="xls",
session=Session(),
tables=[Signature, Signature2])
book_dict = pe.get_book_dict(file_content=test_file.getvalue(),
file_type="xls")
expected = OrderedDict()
expected.update({'signature': [['X', 'Y', 'Z'], [1, 2, 3], [4, 5, 6]]})
expected.update(
{'signature2': [['A', 'B', 'C'], [1, 2, 3], [4, 5, 6]]})
assert book_dict == expected
def setUp(self):
self.testfile = "multiple1.xls"
self.content = OrderedDict()
self.content.update(
{"Sheet1": [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]]})
self.content.update(
{"Sheet2": [[4, 4, 4, 4], [5, 5, 5, 5], [6, 6, 6, 6]]})
self.content.update(
{"Sheet3": [[u'X', u'Y', u'Z'], [1, 4, 7], [2, 5, 8], [3, 6, 9]]})
w = pe.BookWriter(self.testfile)
w.write_book_from_dict(self.content)
w.close()
def test_save_book_as_file_from_sql(self):
test_file = "book_from_sql.xls"
pe.save_book_as(dest_file_name=test_file,
session=Session(),
tables=[Signature, Signature2])
book_dict = pe.get_book_dict(file_name=test_file)
expected = OrderedDict()
expected.update({'signature': [['X', 'Y', 'Z'], [1, 2, 3], [4, 5, 6]]})
expected.update(
{'signature2': [['A', 'B', 'C'], [1, 2, 3], [4, 5, 6]]})
assert book_dict == expected
os.unlink(test_file)
def test_extend_columns2(self):
r = self.testclass(self.testfile)
columns = OrderedDict()
columns.update({"p": ['c1', 'x1']})
columns.update({"a": ['c2', 'x2']})
columns.update({"d": ['c3', 'x4']})
r.extend_columns(columns)
assert r.row[0] == ['a', 'b', 'c', 'd', 'c1', 'c2', 'c3']
assert r.row[1] == ['e', 'f', 'g', 'h', 'x1', 'x2', 'x4']
assert r.row[2] == ['i', 'j', 1.1, 1, '', '', '']
r2 = self.testclass(self.testfile)
columns = OrderedDict()
columns.update({"p": ['c1', 'x1', 'y1', 'z1']})
columns.update({"a": ['c2', 'x2', 'y2']})
columns.update({"d": ['c3', 'x4']})
r2.extend_columns(columns)
assert r2.row[0] == ['a', 'b', 'c', 'd', 'c1', 'c2', 'c3']
assert r2.row[1] == ['e', 'f', 'g', 'h', 'x1', 'x2', 'x4']
assert r2.row[2] == ['i', 'j', 1.1, 1, 'y1', 'y2', '']
assert r2.row[3] == ['', '', '', '', 'z1', '', '']
def test_get_book_from_file(self):
test_file = "test_get_book.xls"
content = OrderedDict()
content.update({"Sheet1": [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]]})
content.update({"Sheet2": [[4, 4, 4, 4], [5, 5, 5, 5], [6, 6, 6, 6]]})
content.update(
{"Sheet3": [[u'X', u'Y', u'Z'], [1, 4, 7], [2, 5, 8], [3, 6, 9]]})
book = pe.Book(content)
book.save_as(test_file)
book2 = pe.get_book(file_name=test_file)
assert book2.to_dict() == content
os.unlink(test_file)
def test_extend_columns2(self):
r = self.testclass(self.testfile)
columns = OrderedDict()
columns.update({"p": ["c1", "x1"]})
columns.update({"a": ["c2", "x2"]})
columns.update({"d": ["c3", "x4"]})
r.extend_columns(columns)
assert r.row[0] == ["a", "b", "c", "d", "c1", "c2", "c3"]
assert r.row[1] == ["e", "f", "g", "h", "x1", "x2", "x4"]
assert r.row[2] == ["i", "j", 1.1, 1, "", "", ""]
r2 = self.testclass(self.testfile)
columns = OrderedDict()
columns.update({"p": ["c1", "x1", "y1", "z1"]})
columns.update({"a": ["c2", "x2", "y2"]})
columns.update({"d": ["c3", "x4"]})
r2.extend_columns(columns)
assert r2.row[0] == ["a", "b", "c", "d", "c1", "c2", "c3"]
assert r2.row[1] == ["e", "f", "g", "h", "x1", "x2", "x4"]
assert r2.row[2] == ["i", "j", 1.1, 1, "y1", "y2", ""]
assert r2.row[3] == ["", "", "", "", "z1", "", ""]
def test_save_book_to_memory_from_sql(self):
test_file = pe.save_book_as(
dest_file_type="xls",
session=Session(),
tables=[Signature, Signature2],
)
book_dict = pe.get_book_dict(file_content=test_file.getvalue(),
file_type="xls")
expected = OrderedDict()
expected.update({"signature": [["X", "Y", "Z"], [1, 2, 3], [4, 5, 6]]})
expected.update(
{"signature2": [["A", "B", "C"], [1, 2, 3], [4, 5, 6]]})
assert book_dict == expected
def test_save_book_as_file_from_sql(self):
test_file = "book_from_sql.xls"
pe.save_book_as(
dest_file_name=test_file,
session=Session(),
tables=[Signature, Signature2],
)
book_dict = pe.get_book_dict(file_name=test_file)
expected = OrderedDict()
expected.update({"signature": [["X", "Y", "Z"], [1, 2, 3], [4, 5, 6]]})
expected.update(
{"signature2": [["A", "B", "C"], [1, 2, 3], [4, 5, 6]]})
assert book_dict == expected
os.unlink(test_file)
def setUp(self):
self.content = OrderedDict()
self.content.update({
"Sheet1": [[u'X', u'Y', u'Z'], [1, 4, 7], [2, 5, 8], [3, 6, 9]]})
self.content.update({
"Sheet2": [[u'A', u'B', u'C'], [1, 4, 7], [2, 5, 8], [3, 6, 9]]})
self.result1 = [
{'Y': 4, 'X': 1, 'Z': 7},
{'Y': 5, 'X': 2, 'Z': 8},
{'Y': 6, 'X': 3, 'Z': 9}]
self.result2 = [
{'B': 4, 'A': 1, 'C': 7},
{'B': 5, 'A': 2, 'C': 8},
{'B': 6, 'A': 3, 'C': 9}]
def test_set_bookdict():
b = Book()
b.bookdict = {"sheet1": [[1]], "sheet2": [[2]]}
expected = dedent("""
sheet1:
+---+
| 1 |
+---+
sheet2:
+---+
| 2 |
+---+""").strip()
eq_(str(b), expected)
expected = OrderedDict([("sheet1", [[1]]), ("sheet2", [[2]])])
eq_(b.bookdict, expected)
def setUp(self):
self.content = OrderedDict([('Sheet 1', [[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0],
[7.0, 8.0, 9.0]]),
('Sheet 2', [['X', 'Y', 'Z'],
[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0]]),
('Sheet 3', [['O', 'P', 'Q'],
[3.0, 2.0, 1.0],
[4.0, 3.0, 2.0]])])
self.testfile = "test.xls"
self.testfile2 = "test.csv"
w = pe.BookWriter(self.testfile)
w.write_book_from_dict(self.content)
w.close()
def setUp(self):
self.content = OrderedDict()
self.content.update(
{"Sheet1": [[u"X", u"Y", u"Z"], [1, 4, 7], [2, 5, 8], [3, 6, 9]]})
self.content.update(
{"Sheet2": [[u"A", u"B", u"C"], [1, 4, 7], [2, 5, 8], [3, 6, 9]]})
self.result1 = [
{
"Y": 4,
"X": 1,
"Z": 7
},
{
"Y": 5,
"X": 2,
"Z": 8
},
{
"Y": 6,
"X": 3,
"Z": 9
},
]
self.result2 = [
{
"B": 4,
"A": 1,
"C": 7
},
{
"B": 5,
"A": 2,
"C": 8
},
{
"B": 6,
"A": 3,
"C": 9
},
]
def test_add(self):
s = Sheet(self.data, "test")
s.name_rows_by_column(0)
data = OrderedDict({"Row 5": [10, 11, 12]})
s = s.row + data
assert s.row["Row 5"] == [10, 11, 12]
def test_add(self):
s = Sheet(self.data, "test")
s.name_columns_by_row(2)
data = OrderedDict({"Column 4": [10, 11, 12]})
s = s.column + data
assert s.column["Column 4"] == [10, 11, 12]
def test_add(self):
s = Sheet(self.data, "test")
s.name_columns_by_row(0)
data = OrderedDict({"Column 4": [10, 11, 12]})
s = s.column + data
eq_(s.column.Column_4, [10, 11, 12])
