def test_add_textfield_bad_template(self):
tc = Thinkcell()
template = "example.pptx"
with pytest.raises(ValueError) as e_info:
tc.add_textfield(
template_name="example2.pptx",
field_name="Title",
text="A great slide",
)
def test_add_chart_bad_template(self):
tc = Thinkcell()
template = "example.pptx"
with pytest.raises(ValueError) as e_info:
tc.add_chart(
template_name="example2.pptx",
chart_name="Cool Name bro",
categories=["Alpha", "bravo"],
data=[[3, 4, datetime(2012, 9, 16, 0, 0)], [2, "adokf", 6]],
)
def test_save_ppttc_bad_file(self, input, output):
tc = Thinkcell()
tc.add_template("example.pptx")
tc.add_chart(
template_name="example.pptx",
chart_name="Chart name",
categories=["alpha", "bravo"],
data=[["today", 1, 2], ["tomorrow", 3, 4]],
)
with pytest.raises(output) as e_info:
tc.save_ppttc(filename=input)
def test_save_ppttc(self):
tc = Thinkcell()
tc.add_template("example.pptx")
tc.add_chart(
template_name="example.pptx",
chart_name="Chart name",
categories=["alpha", "bravo"],
data=[["today", 1, 2], ["tomorrow", 3, 4]],
)
assert tc.save_ppttc(filename="test.ppttc") == True
os.remove("test.ppttc")
def test_add_textfield_warning(self):
tc = Thinkcell()
template_name = "template.pptx"
tc.add_template(template_name)
with pytest.warns(UserWarning) as record:
tc.add_textfield(
template_name=template_name,
field_name=234,
text="A great slide",
)
def test_add_chart_warning(self):
tc = Thinkcell()
template_name = "template.pptx"
tc.add_template(template_name)
with pytest.warns(UserWarning) as record:
tc.add_chart(
template_name=template_name,
chart_name=234,
categories=["Alpha", "bravo"],
data=[[3, 4, datetime(2012, 9, 16, 0, 0)], [2, "adokf", 6]],
)
def test_add_chart_from_dataframe_invalid_dataframe(self):
tc = Thinkcell()
template = "example.pptx"
dataframe = [["Apple", 200, 1.5, 10], ["Amazon", 100, 1.0, 12],
["Slack", 50, 0.5, 16]]
tc.add_template(template)
with pytest.raises(DataFrameError) as e_info:
tc.add_chart_from_dataframe(
template_name=template,
chart_name="Cool Chart",
dataframe=dataframe,
)
def test_add_chart_from_dataframe_no_rows(self):
tc = Thinkcell()
template = "example.pptx"
dataframe = pd.DataFrame(
columns=["Company", "Employees", "Revenue", "Other"], )
tc.add_template(template)
with pytest.raises(DataFrameError) as e_info:
tc.add_chart_from_dataframe(
template_name=template,
chart_name="Cool Chart",
dataframe=dataframe,
)
def test_add_chart_from_dataframe_no_data(self):
tc = Thinkcell()
template = "example.pptx"
dataframe = pd.DataFrame(
columns=["Company"],
data=[["Apple"], ["Amazon"], ["Slack"]],
)
tc.add_template(template)
with pytest.raises(DataFrameError) as e_info:
tc.add_chart_from_dataframe(
template_name=template,
chart_name="Cool Chart",
dataframe=dataframe,
)
def test_add_chart(self):
tc = Thinkcell()
template = "example.pptx"
tc.add_template(template)
tc.add_chart(
template_name=template,
chart_name="Cool Name bro",
categories=["Alpha", "bravo"],
data=[[3, 4, datetime(2012, 9, 16, 0, 0)], [2, "adokf", 4]],
)
assert tc.charts == [{
"template":
"example.pptx",
"data": [{
"name":
"Cool Name bro",
"table": [
[None, {
"string": "Alpha"
}, {
"string": "bravo"
}],
[],
[
{
"number": 3
},
{
"number": 4
},
{
"date": "2012-09-16"
},
],
[
{
"number": 2
},
{
"string": "adokf"
},
{
"number": 4
},
],
],
}],
}]
def test_add_textfield(self):
tc = Thinkcell()
template = "example.pptx"
tc.add_template(template)
tc.add_textfield(
template_name=template,
field_name="Title",
text="A great slide",
)
assert tc.charts == [{
"template":
"example.pptx",
"data": [{
"name": "Title",
"table": [
[
{
"string": "A great slide"
},
],
],
}],
}]
def test_add_template(self):
tc = Thinkcell()
template = "example.pptx"
tc.add_template(template)
assert tc.charts == [{"template": template, "data": []}]
def test_verify_template_3(self):
template_name = "example.pptx"
assert Thinkcell.verify_template(template_name) == template_name
def test_verify_template_2(self):
template_name = 5
with pytest.raises(TypeError) as e_info:
Thinkcell.verify_template(template_name)
def test_verify_template_1(self):
template_name = "not a file name"
with pytest.raises(TypeError) as e_info:
Thinkcell.verify_template(template_name)
chart_4 = {
"chart_name":
"Chart4",
"dataframe":
pd.DataFrame(
columns=["Metric", 2017, 2018, 2019],
data=[["Expenses", 10, 10, 50], ["Revenues", 1, 1, 40]],
),
}
charts = [chart_1, chart_2, chart_3]
template_name = "complex-template-text-field.pptx"
filename = "complex-example-with-multiple-slides.ppttc"
tc = Thinkcell()
tc.add_template(template_name)
for chart in charts:
tc.add_chart_from_dataframe(
template_name=template_name,
chart_name=chart["chart_name"],
dataframe=chart["dataframe"],
)
tc.add_textfield(template_name=template_name,
field_name="chart_text",
text="First instance")
charts = [chart_1, chart_2, chart_4]
tc.add_template(template_name)
def test_transform_input(self, test_input, expected):
assert Thinkcell.transform_input(test_input) == expected
def test_str_all(self):
tc = Thinkcell()
assert str(tc) == "[]"
def test_init(self):
tc = Thinkcell()
assert tc.charts == []
def test_transform_input_bad(self):
with pytest.raises(ValueError) as e_info:
Thinkcell.transform_input([3, 4])
def test_save_pptc(self):
tc = Thinkcell()
with pytest.raises(ValueError) as e_info:
tc.save_ppttc("test.ppttc")
def test_add_chart_from_dataframe(self):
tc = Thinkcell()
template = "example.pptx"
dataframe = pd.DataFrame(
columns=["Company", "Employees", "Revenue", "Other"],
data=[["Apple", 200, 1.5, 10], ["Amazon", 100, 1.0, 12],
["Slack", 50, 0.5, 16]],
)
tc.add_template(template)
tc.add_chart_from_dataframe(
template_name=template,
chart_name="Cool Chart",
dataframe=dataframe,
)
assert tc.charts == [
{
"template":
"example.pptx",
"data": [
{
"name":
"Cool Chart",
"table": [
[
None, {
"string": "Employees"
}, {
"string": "Revenue"
}, {
"string": "Other"
}
],
[],
[
{
"string": "Apple"
},
{
"number": 200
},
{
"number": 1.5
},
{
"number": 10
},
],
[
{
"string": "Amazon"
},
{
"number": 100
},
{
"number": 1.0
},
{
"number": 12
},
],
[
{
"string": "Slack"
},
{
"number": 50
},
{
"number": 0.5
},
{
"number": 16
},
],
],
},
],
},
]
from thinkcell import Thinkcell
template_name = "simple-template.pptx"
categories = ["Ads", "iPhones", "Other"]
chart_name = "Chart1"
filename = "simple-example.ppttc"
data = [["Apple", 1, 11, 14], ["Google", 8, 2, 15], ["Microsoft", 1, 2, 12]]
tc = Thinkcell()
tc.add_template(template_name)
tc.add_chart(
template_name=template_name,
chart_name=chart_name,
categories=categories,
data=data,
)
tc.save_ppttc(filename=filename)
"Chart2",
"data": [
["Manchester", 34, 20, 14],
["Chelsea", 40, 13, 20],
["Arsenal", 34, 30, 4],
],
}
chart_3 = {
"categories": [2017, 2018, 2019],
"chart_name": "Chart3",
"data": [["Revenue", 45, 24, 0], ["Employees", 10, 14, 0]],
}
charts = [chart_1, chart_2, chart_3]
template_name = "complex-template.pptx"
filename = "complex-example.ppttc"
tc = Thinkcell()
tc.add_template(template_name)
for chart in charts:
tc.add_chart(
template_name=template_name,
chart_name=chart["chart_name"],
categories=chart["categories"],
data=chart["data"],
)
tc.save_ppttc(filename=filename)
