def test_dict_to_html():
from bs4 import BeautifulSoup
from collections import OrderedDict
data = OrderedDict()
data['A'] = 1
data['B'] = 2
data['C'] = [{'D': 3}, {'E': 4}]
html = dict_to_html(data)
soup = BeautifulSoup(html, 'html.parser')
tr = soup.find_all('tr')
sub_tr = tr[2].find_all('tr')
actual = (len(tr), len(sub_tr))
expected = (6, 3)
assert actual == expected
def _repr_html_(self):
df = self.get_document_information()
columns = df.columns.tolist()
dict_info = df.set_index(columns[1]).to_dict()
info = None
header = None
for key, value in dict_info.items():
if key != 'label_ko':
info = value
keys = list(info.keys())
header = ['', keys[0]]
info.pop(keys[0])
df = self.get_entity_information()
columns = df.columns.tolist()
df = df.drop(columns[0], axis=1).set_index(columns[1])
info['Company name'] = df.loc['Entity iegistrant name'][0]
return dict_to_html(info, header=header)
def _repr_html_(self):
return dict_to_html(self.to_dict(),
exclude=['pages'],
header=['Label', 'Data'])
def _repr_html_(self) -> str:
return dict_to_html(self.to_dict(), header=['Label', 'Data'])
def _repr_html_(self):
return dict_to_html(self.to_dict(summary=False), header=['Label', 'Data'])