def test_write_a_book_reader(self):
b = pe.BookReader(self.testfile)
b.save_as(self.testfile2)
x = pe.BookReader(self.testfile2)
for s in x:
data = pe.utils.to_array(s)
assert self.content[s.name] == data
def test_write_book_reader(self):
reader = pe.BookReader(self.testfile)
writer = pe.BookWriter(self.testfile2)
writer.write_book_reader(reader)
writer.close()
reader2 = pe.BookReader(self.testfile2)
data = pe.utils.to_dict(reader2)
assert data == self.content
def test_add_book3(self):
"""
test this scenario: book3 = sheet1 + sheet2
"""
b1 = pyexcel.BookReader(self.testfile, library="pyexcel-odsr")
b2 = pyexcel.BookReader(self.testfile2)
b3 = b1["Sheet1"] + b2["Sheet3"]
content = b3.dict
sheet_names = content.keys()
assert len(sheet_names) == 2
assert content["Sheet3"] == self.content["Sheet3"]
assert content["Sheet1"] == self.content["Sheet1"]
def test_add_book3(self):
"""
test this scenario: book3 = sheet1 + sheet2
"""
b1 = pe.BookReader(self.testfile)
b2 = pe.BookReader(self.testfile2)
b3 = b1["Sheet1"] + b2["Sheet3"]
content = pe.utils.to_dict(b3)
sheet_names = content.keys()
assert len(sheet_names) == 2
assert content["Sheet3"] == self.content["Sheet3"]
assert content["Sheet1"] == self.content["Sheet1"]
def test_write_a_book_reader(self):
b = pe.BookReader(self.testfile)
bw = pe.BookWriter(self.testfile2)
for s in b:
data = pe.utils.to_array(s)
sheet = bw.create_sheet(s.name)
sheet.write_array(data)
sheet.close()
bw.close()
x = pe.BookReader(self.testfile2)
for s in x:
data = pe.utils.to_array(s)
assert self.content[s.name] == data
def test_add_sheet(self):
"""
test this scenario: book3 = sheet1 + single_sheet_book
"""
b1 = pe.BookReader(self.testfile)
b2 = pe.BookReader(self.test_single_sheet_file)
b3 = b1["Sheet1"] + b2
content = pe.utils.to_dict(b3)
sheet_names = content.keys()
assert len(sheet_names) == 2
for name in sheet_names:
if "Sheet1" in name:
assert content[name] == self.content["Sheet1"]
elif "single.xls" in name:
assert content[name] == self.content1["Sheet1"]
def test_merge_any_files_to_a_book(self):
file_array = [
self.testfile,
self.testfile2,
self.testfile3,
self.testfile4,
]
pe.cookbook.merge_all_to_a_book(file_array, "merged.xlsx")
r = pe.BookReader("merged.xlsx")
r[self.testfile].name_columns_by_row(0)
content = r[self.testfile].to_dict()
assert content == self.content
r[self.testfile2].format(int)
r[self.testfile2].name_columns_by_row(0)
content2 = r[self.testfile2].to_dict()
assert content2 == self.content2
r[self.testfile3].name_columns_by_row(0)
content3 = r[self.testfile3].to_dict()
assert content3 == self.content3
content4 = r["Sheet1"].to_array()
assert content4 == self.content4["Sheet1"]
content5 = r["Sheet2"].to_array()
assert content5 == self.content4["Sheet2"]
content6 = r["Sheet3"].to_array()
assert content6 == self.content4["Sheet3"]
def test_add_book4(self):
"""
test this scenario: book3 = sheet1 + book
"""
b1 = pyexcel.BookReader(self.testfile, library="pyexcel-odsr")
b2 = pyexcel.BookReader(self.testfile2)
b3 = b1["Sheet1"] + b2
content = b3.dict
sheet_names = content.keys()
assert len(sheet_names) == 4
for name in sheet_names:
if "Sheet3" in name:
assert content[name] == self.content["Sheet3"]
elif "Sheet2" in name:
assert content[name] == self.content["Sheet2"]
elif "Sheet1" in name:
assert content[name] == self.content["Sheet1"]
def test_add_sheet2(self):
"""
test this scenario: book3 = sheet1 + single_sheet_book
"""
b1 = pe.BookReader(self.testfile)
b3 = b1["Sheet1"] + b1["Sheet1"]
b3["Sheet1"].row[0] = [3, 3, 3, 3]
eq_(b1["Sheet1"].row[0], [1, 1, 1, 1])
def test_random_access_operator(self):
r = pyexcel.BookReader(self.testfile)
value = r["Sheet1"][0, 1]
assert value == 1
value = r["Sheet3"][0, 1]
assert value == 'Y'
r["Sheet3"].name_columns_by_row(0)
assert r["Sheet3"][0, 1] == 4
def test_add_book2_in_place(self):
"""
test this scenario: book3 = book1 + sheet3
"""
b1 = pe.BookReader(self.testfile)
b2 = pe.BookReader(self.testfile2)
b1 += b2["Sheet3"]
content = pe.utils.to_dict(b1)
sheet_names = content.keys()
assert len(sheet_names) == 4
for name in sheet_names:
if "Sheet3" in name:
assert content[name] == self.content["Sheet3"]
elif "Sheet2" in name:
assert content[name] == self.content["Sheet2"]
elif "Sheet1" in name:
assert content[name] == self.content["Sheet1"]
def test_add_book1_in_place(self):
"""
test this scenario: book1 += book2
"""
b1 = pyexcel.BookReader(self.testfile)
b2 = pyexcel.BookReader(self.testfile2)
b1 += b2
content = b1.dict
sheet_names = content.keys()
assert len(sheet_names) == 6
for name in sheet_names:
if "Sheet3" in name:
assert content[name] == self.content["Sheet3"]
elif "Sheet2" in name:
assert content[name] == self.content["Sheet2"]
elif "Sheet1" in name:
assert content[name] == self.content["Sheet1"]
def test_add_book2(self):
"""
test this scenario: book3 = book1 + sheet3
"""
b1 = pyexcel.BookReader(self.testfile)
b2 = pyexcel.BookReader(self.testfile2)
b3 = b1 + b2["Sheet3"]
content = b3.dict
sheet_names = content.keys()
assert len(sheet_names) == 4
for name in sheet_names:
if "Sheet3" in name:
assert content[name] == self.content["Sheet3"]
elif "Sheet2" in name:
assert content[name] == self.content["Sheet2"]
elif "Sheet1" in name:
assert content[name] == self.content["Sheet1"]
def test_add_book4(self):
"""
test this scenario: book3 = sheet1 + book
"""
b1 = pe.BookReader(self.testfile)
b2 = pe.BookReader(self.testfile2)
b3 = b1["Sheet1"] + b2
content = pe.utils.to_dict(b3)
sheet_names = content.keys()
assert len(sheet_names) == 4
for name in sheet_names:
if "Sheet3" in name:
assert content[name] == self.content["Sheet3"]
elif "Sheet2" in name:
assert content[name] == self.content["Sheet2"]
elif "Sheet1" in name:
assert content[name] == self.content["Sheet1"]
def test_iterate_through_sheets(self):
b = pyexcel.BookReader(self.testfile)
for s in b:
data = pyexcel.utils.to_array(s)
assert self.content[s.name] == data
si = pyexcel.iterators.SheetIterator(b)
for s in si:
data = pyexcel.utils.to_array(s)
assert self.content[s.name] == data
def test_random_access_operator(self):
r = pe.BookReader(self.testfile)
value = r["Sheet1"].row[0][1]
assert value == 1
value = r["Sheet3"].row[0][1]
assert value == 'Y'
r["Sheet3"].name_columns_by_row(0)
value = r["Sheet3"].row[0][1]
assert value == 4
def test_add_book6(self):
"""
test this scenario: book3 = book + single_sheet_book
"""
b1 = pe.BookReader(self.test_single_sheet_file)
b2 = pe.BookReader(self.testfile2)
b3 = b2 + b1
content = b3.dict
sheet_names = content.keys()
assert len(sheet_names) == 4
for name in sheet_names:
if "Sheet3" in name:
assert content[name] == self.content["Sheet3"]
elif "Sheet2" in name:
assert content[name] == self.content["Sheet2"]
elif "Sheet1" in name:
assert content[name] == self.content["Sheet1"]
elif "single.xls" in name:
assert content[name] == self.content1["Sheet1"]
def test_add_book4_2(self):
"""
test this scenario: book3 = sheet1 + book
use . notation
"""
b1 = pe.BookReader(self.testfile)
b2 = pe.BookReader(self.testfile2)
b3 = b1.Sheet1 + b2
content = b3.dict
sheet_names = content.keys()
assert len(sheet_names) == 4
for name in sheet_names:
if "Sheet3" in name:
assert content[name] == self.content["Sheet3"]
elif "Sheet2" in name:
assert content[name] == self.content["Sheet2"]
elif "Sheet1" in name:
assert content[name] == self.content["Sheet1"]
def test_merge_csv_files_to_a_book(self):
file_array = [self.testfile, self.testfile2, self.testfile3]
pe.cookbook.merge_csv_to_a_book(file_array, "merged.xlsx")
r = pe.BookReader("merged.xlsx")
content = pe.utils.to_dict(r[self.testfile].become_series())
assert content == self.content
content2 = pe.utils.to_dict(r[self.testfile2].become_series())
assert content2 == self.content2
content3 = pe.utils.to_dict(r[self.testfile3].become_series())
assert content3 == self.content3
def test_random_access_operator(self):
r = pe.BookReader(self.testfile)
value = r["Sheet1"].row[0][1]
assert value == 1
value = r["Sheet3"].row[0][1]
assert value == 'Y'
value = r["Sheet3"].become_series().row[0][1]
assert value == 4
value = r["Sheet3"].become_sheet().row[0][1]
assert value == 'Y'
def test_add_sheet2(self):
"""
test this scenario: book3 = sheet1 + single_sheet_book
"""
b1 = pe.BookReader(self.testfile)
b3 = b1["Sheet1"] + b1["Sheet1"]
content = b3.dict
sheet_names = content.keys()
assert len(sheet_names) == 2
for name in sheet_names:
if "Sheet1" in name:
assert content[name] == self.content["Sheet1"]
def test_merge_csv_files_to_a_book(self):
file_array = [self.testfile, self.testfile2, self.testfile3]
pe.cookbook.merge_csv_to_a_book(file_array, "merged.xlsx")
r = pe.BookReader("merged.xlsx")
r[self.testfile].name_columns_by_row(0)
content = r[self.testfile].to_dict()
assert content == self.content
r[self.testfile2].format(int)
r[self.testfile2].name_columns_by_row(0)
content2 = r[self.testfile2].to_dict()
assert content2 == self.content2
r[self.testfile3].name_columns_by_row(0)
content3 = r[self.testfile3].to_dict()
assert content3 == self.content3
def test_add_book_error(self):
"""
test this scenario: book3 = sheet1 + book
"""
b1 = pyexcel.BookReader(self.testfile)
try:
b1 + 12
assert 1 == 2
except TypeError:
assert 1 == 1
try:
b1 += 12
assert 1 == 2
except TypeError:
assert 1 == 1
def test_add_book_error(self):
"""
test this scenario: book3 = book + integer
"""
b1 = pe.BookReader(self.testfile)
try:
b1 + 12
assert 1==2
except TypeError:
assert 1==1
try:
b1 += 12
assert 1==2
except TypeError:
assert 1==1
def test_reading_through_sheets(self):
b = pyexcel.BookReader(self.testfile)
data = list(b["Sheet1"].rows())
expected = [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]]
assert data == expected
data = list(b["Sheet2"].rows())
expected = [[4, 4, 4, 4], [5, 5, 5, 5], [6, 6, 6, 6]]
assert data == expected
data = list(b["Sheet3"].rows())
expected = [[u"X", u"Y", u"Z"], [1, 4, 7], [2, 5, 8], [3, 6, 9]]
assert data == expected
sheet3 = b["Sheet3"]
sheet3.name_columns_by_row(0)
data = list(b["Sheet3"].rows())
expected = [[1, 4, 7], [2, 5, 8], [3, 6, 9]]
assert data == expected
def test_reading_through_sheets(self):
b = pe.BookReader(self.testfile)
data = pe.utils.to_array(b["Sheet1"].rows())
expected = [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]]
assert data == expected
data = pe.utils.to_array(b["Sheet2"].rows())
expected = [[4, 4, 4, 4], [5, 5, 5, 5], [6, 6, 6, 6]]
assert data == expected
data = pe.utils.to_array(b["Sheet3"].rows())
expected = [[u'X', u'Y', u'Z'], [1, 4, 7], [2, 5, 8], [3, 6, 9]]
assert data == expected
sheet3 = b["Sheet3"]
sheet3.become_series()
data = pe.utils.to_array(b["Sheet3"].rows())
expected = [[1, 4, 7], [2, 5, 8], [3, 6, 9]]
assert data == expected
def test_write_book_reader(self):
reader = pe.BookReader(self.testfile)
writer = pe.BookWriter(self.testfile2)
writer.write_book_reader(reader)
writer.close()
sheet1 = "test_Sheet 1.csv"
reader1 = pe.Reader(sheet1)
data = pe.utils.to_array(reader1)
assert data == self.content["Sheet 1"]
sheet2 = "test_Sheet 2.csv"
reader2 = pe.Reader(sheet2)
data = pe.utils.to_array(reader2)
assert data == self.content["Sheet 2"]
sheet3 = "test_Sheet 3.csv"
reader3 = pe.Reader(sheet3)
data = pe.utils.to_array(reader3)
assert data == self.content["Sheet 3"]
def test_merge_any_files_to_a_book(self):
file_array = [
self.testfile, self.testfile2, self.testfile3, self.testfile4
]
pe.cookbook.merge_all_to_a_book(file_array, "merged.xlsx")
r = pe.BookReader("merged.xlsx")
content = pe.utils.to_dict(r[self.testfile].become_series())
assert content == self.content
content2 = pe.utils.to_dict(r[self.testfile2].become_series())
assert content2 == self.content2
content3 = pe.utils.to_dict(r[self.testfile3].become_series())
assert content3 == self.content3
content4 = pe.utils.to_array(r["Sheet1"])
assert content4 == self.content4["Sheet1"]
content5 = pe.utils.to_array(r["Sheet2"])
assert content5 == self.content4["Sheet2"]
content6 = pe.utils.to_array(r["Sheet3"])
assert content6 == self.content4["Sheet3"]
def test_add_sheet_error(self):
"""
test this scenario: book3 = sheet1 + integer
"""
b1 = pe.BookReader(self.testfile)
try:
b1["Sheet1"] + 12
assert 1==2
except TypeError:
assert 1==1
try:
b1["Sheet1"] += 12
assert 1==2
except TypeError:
assert 1==1
try:
b1["Sheet1"] += pe.sheets.AS_COLUMNS(12)
assert 1==2
except TypeError:
assert 1==1
def test_write_book_reader(self):
reader = pe.BookReader(self.testfile)
writer = pe.BookWriter(self.testfile2)
writer.write_book_reader(reader)
writer.close()
sheet1 = "test__Sheet 1__0.csv"
reader1 = pe.Reader(sheet1)
reader1.apply_formatter(pe.formatters.SheetFormatter(float))
data = pe.utils.to_array(reader1)
assert data == self.content["Sheet 1"]
sheet2 = "test__Sheet 2__1.csv"
reader2 = pe.Reader(sheet2)
reader2.apply_formatter(pe.formatters.SheetFormatter(float))
data = pe.utils.to_array(reader2)
assert data == self.content["Sheet 2"]
sheet3 = "test__Sheet 3__2.csv"
reader3 = pe.Reader(sheet3)
reader3.apply_formatter(pe.formatters.SheetFormatter(float))
data = pe.utils.to_array(reader3)
assert data == self.content["Sheet 3"]
