def test_register_writer(self):
# some awkward mocking to test out dispatch and such actually works
called_save = []
called_write_cells = []
class DummyClass(ExcelWriter):
called_save = False
called_write_cells = False
supported_extensions = ['xlsx', 'xls']
engine = 'dummy'
def save(self):
called_save.append(True)
def write_cells(self, *args, **kwargs):
called_write_cells.append(True)
def check_called(func):
func()
assert len(called_save) >= 1
assert len(called_write_cells) >= 1
del called_save[:]
del called_write_cells[:]
with pd.option_context('io.excel.xlsx.writer', 'dummy'):
register_writer(DummyClass)
writer = ExcelWriter('something.xlsx')
assert isinstance(writer, DummyClass)
df = tm.makeCustomDataframe(1, 1)
check_called(lambda: df.to_excel('something.xlsx'))
check_called(
lambda: df.to_excel(
'something.xls', engine='dummy'))
def test_register_writer(self):
# some awkward mocking to test out dispatch and such actually works
called_save = []
called_write_cells = []
class DummyClass(ExcelWriter):
called_save = False
called_write_cells = False
supported_extensions = ["xlsx", "xls"]
engine = "dummy"
def save(self):
called_save.append(True)
def write_cells(self, *args, **kwargs):
called_write_cells.append(True)
def check_called(func):
func()
assert len(called_save) >= 1
assert len(called_write_cells) >= 1
del called_save[:]
del called_write_cells[:]
with pd.option_context("io.excel.xlsx.writer", "dummy"):
register_writer(DummyClass)
writer = ExcelWriter("something.xlsx")
assert isinstance(writer, DummyClass)
df = tm.makeCustomDataframe(1, 1)
check_called(lambda: df.to_excel("something.xlsx"))
check_called(lambda: df.to_excel("something.xls", engine="dummy"))
def test_table(self):
max_n = 3
min_n = 2
ngram_multiplier = 4
num_ngrams_report = 25
num_ngrams_wordcloud = 25
num_ngrams = max(num_ngrams_report, num_ngrams_wordcloud)
tfidf_cold = TFIDF(ReferenceData.cold_df,
tokenizer=LemmaTokenizer(),
ngram_range=(min_n, max_n))
tfidf_random = TFIDF(ReferenceData.random_df,
tokenizer=LemmaTokenizer(),
ngram_range=(min_n, max_n))
citation_count_dict = {
1: 10,
2: 3,
101: 2,
102: 0,
103: 5,
104: 4,
105: 10
}
pick = 'sum'
time = False
focus = 'chi2'
register_writer(TestTableOutput.FakeWriter)
fake_writer = TestTableOutput.FakeWriter('spreadsheet.fake')
table_output(tfidf_cold, tfidf_random, citation_count_dict, num_ngrams,
pick, ngram_multiplier, time, focus, fake_writer)
# Check sheet headings...
self.assertListEqual([
None, 'Term', 'Score', 'Rank', 'Focus chi2 Score',
'Focus chi2 Rank', 'Diff Base to Focus Rank', 'Time Score',
'Time Rank', 'Diff Base to Time Rank', 'Citation Score',
'Citation Rank', 'Diff Base to Citation Rank'
], fake_writer.sheets['Summary'][0])
self.assertListEqual([None, 'Term', 'Score', 'Rank'],
fake_writer.sheets['Base'][0])
self.assertListEqual(
[None, 'Term', 'Focus chi2 Score', 'Focus chi2 Rank'],
fake_writer.sheets['Focus'][0])
self.assertListEqual([None, 'Term', 'Time Score', 'Time Rank'],
fake_writer.sheets['Time'][0])
self.assertListEqual([None, 'Term', 'Citation Score', 'Citation Rank'],
fake_writer.sheets['Cite'][0])
# Base sheet should match summary sheet
for y in range(25):
for x in range(4):
self.assertEqual(fake_writer.sheets['Summary'][y + 1][x],
fake_writer.sheets['Base'][y + 1][x])
def test_register_writer(self):
class DummyClass(ExcelWriter):
called_save = False
called_write_cells = False
called_sheets = False
_supported_extensions = ("xlsx", "xls")
_engine = "dummy"
def book(self):
pass
def _save(self):
type(self).called_save = True
def _write_cells(self, *args, **kwargs):
type(self).called_write_cells = True
@property
def sheets(self):
type(self).called_sheets = True
@classmethod
def assert_called_and_reset(cls):
assert cls.called_save
assert cls.called_write_cells
assert not cls.called_sheets
cls.called_save = False
cls.called_write_cells = False
register_writer(DummyClass)
with option_context("io.excel.xlsx.writer", "dummy"):
path = "something.xlsx"
with tm.ensure_clean(path) as filepath:
with ExcelWriter(filepath) as writer:
assert isinstance(writer, DummyClass)
df = tm.makeCustomDataframe(1, 1)
df.to_excel(filepath)
DummyClass.assert_called_and_reset()
with tm.ensure_clean("something.xls") as filepath:
df.to_excel(filepath, engine="dummy")
DummyClass.assert_called_and_reset()
def test_register_writer(self):
# some awkward mocking to test out dispatch and such actually works
called_save = []
called_write_cells = []
class DummyClass(ExcelWriter):
called_save = False
called_write_cells = False
supported_extensions = ['test', 'xlsx', 'xls']
engine = 'dummy'
def save(self):
called_save.append(True)
def write_cells(self, *args, **kwargs):
called_write_cells.append(True)
def check_called(func):
func()
self.assert_(len(called_save) >= 1)
self.assert_(len(called_write_cells) >= 1)
del called_save[:]
del called_write_cells[:]
register_writer(DummyClass)
writer = ExcelWriter('something.test')
tm.assert_isinstance(writer, DummyClass)
df = tm.makeCustomDataframe(1, 1)
panel = tm.makePanel()
func = lambda: df.to_excel('something.test')
check_called(func)
check_called(lambda: panel.to_excel('something.test'))
from pandas import set_option, get_option
val = get_option('io.excel.xlsx.writer')
set_option('io.excel.xlsx.writer', 'dummy')
check_called(lambda: df.to_excel('something.xlsx'))
check_called(lambda: df.to_excel('something.xls', engine='dummy'))
set_option('io.excel.xlsx.writer', val)
def test_register_writer(self):
# some awkward mocking to test out dispatch and such actually works
called_save = []
called_write_cells = []
class DummyClass(ExcelWriter):
called_save = False
called_write_cells = False
supported_extensions = ['test', 'xlsx', 'xls']
engine = 'dummy'
def save(self):
called_save.append(True)
def write_cells(self, *args, **kwargs):
called_write_cells.append(True)
def check_called(func):
func()
self.assert_(len(called_save) >= 1)
self.assert_(len(called_write_cells) >= 1)
del called_save[:]
del called_write_cells[:]
register_writer(DummyClass)
writer = ExcelWriter('something.test')
tm.assert_isinstance(writer, DummyClass)
df = tm.makeCustomDataframe(1, 1)
panel = tm.makePanel()
func = lambda: df.to_excel('something.test')
check_called(func)
check_called(lambda: panel.to_excel('something.test'))
from pandas import set_option, get_option
val = get_option('io.excel.xlsx.writer')
set_option('io.excel.xlsx.writer', 'dummy')
check_called(lambda: df.to_excel('something.xlsx'))
check_called(lambda: df.to_excel('something.xls', engine='dummy'))
set_option('io.excel.xlsx.writer', val)
def test_table(self):
cold_abstracts = [
('1', '''A refrigerator including a main body provided with a refrigerating chamber at an upper
section and with a freezing chamber at a lower section, an ice making tray disposed in an upper space
of an ice making chamber ined in the refrigerating chamber, a first storage container disposed in a
lower space of the ice making chamber to store ice falling down from the ice making tray, and a second
storage container disposed in a freezing chamber to store ice transferred from the ice making tray. The
main body includes a guide channel to guide, when the first storage container reaches an ice-full
state, ice falling from the ice making tray to the second storage container in the freezing chamber.
The size of the ice making chamber is greatly reduced while a sufficient amount of the ice may be
stored, thus securing a larger available space in the refrigerating chamber.'''),
('2', '''The invention provides methods and compositions for maintaining a refrigerating chamber at a
constant temperature. This will maintain a quantity of ice stored ready for consumption.'''),
('3', '''An ice piece release system that includes a chilled compartment set at a temperature below
0.degree. C., a warm section at a temperature above 0.degree. C., and a tray in thermal communication
with the chilled compartment. The tray includes a plurality of ice piece-forming receptacles and a
cavity in thermal communication with the receptacles. The ice piece release system also includes a
primary reservoir assembly in thermal communication with the warm section and fluid communication with
the cavity of the tray. The ice piece release system further includes a heat-exchanging fluid having a
freezing point below that of water, and the fluid resides in the primary reservoir assembly and the
cavity of the tray. The primary reservoir assembly is further adapted to move at least a portion of the
heat-exchanging fluid in the reservoir assembly into the cavity.'''),
('4', '''A refrigerator, in particular a household refrigerator, includes an utility chamber for cooled
goods and a control device, with which a cold air flow can be introduced into the utility chamber when
a cooling signal is present. A defrost heating element is rendered operative by the control device to
prevent the formation of condensate and/or ice due to the cold air flow fed into the utility chamber.
A timing element keeps the heating element out of operation for a predetermined time interval in
response to the generation of the cooling signal.''')
]
random_abstracts = [
('101', '''Acoustic volume indicators for determining liquid or gas volume within a container comprise a
contactor to vibrate a container wall, a detector to receive vibration data from the container wall,
a processor to convert vibration data to frequency information and compare the frequency information to
characteristic container frequency vs. volume data to obtain the measured volume, and an indicator for
displaying the measured volume. The processor may comprise a microprocessor disposed within a housing
having lights that each represent a particular volume. The microprocessor is calibrated to provide an
output signal to a light that indicates the container volume. The processor may comprise a computer and
computer program that converts the data to frequency information, analyzes the frequency information to
identify a peak frequency, compares the peak frequency to the characteristic frequency vs. volume data
to determine the measured volume, and displays the measured volume on a video monitor. '''),
('102', '''A single-module deployable bolted flange connection apparatus ( 10 ) makes up standard flange
joints ( 24, 32 ) for various pipeline tie-in situations, such as spool piece connection and
flowline-tree connections, without the use of divers and auxiliary multiple pieces of equipment. An
outer Flange Alignment Frame (FAF) ( 14 ), carries one or more claws ( 38 ) for grabbing the pipe/spool
to provide flange alignment. The claws are suspended and driven by a novel arrangement of five
hydraulic rams ( 412 - 420 ) A crash-resistant inner frame ( 148 ) houses complete connection tooling
( 150, 152 etc.) The tooling performs the final alignment steps, inserts the gasket and studs, applies
the required tension, and connects the nuts. Studs and nuts are stored separately from the tooling in
an indexed carou, to permit multiple operations, reverse operations (disconnection), and re-work of
failed steps, all without external intervention. '''),
('103', '''A passenger seat with increased knee space for an aft-seated passenger, including a seat base for
being attached to a supporting deck and at least one seat frame including a seat back and seat bottom
carried by the seat base. At least one arm rest assembly is carried by the seat frame and including an
arm rest mounted for pivotal movement about a pivot member between a use position with an upper support
surface in a horizontal position for supporting a forearm of a passenger seated in the seat, and a
stowed position wherein the upper support surface of the arm rest is perpendicular to the use position.
The arm rest pivot member is mounted on the seat frame at a point forward of a plane ined by the seat
back carried by the seat and above a point ined by the seat bottom for allowing the knee of an
aft-seated passenger to occupy space behind the pivot member of the arm rest. '''),
('104', '''A bag having two accesses, an inlet and an outlet, both containing closing devices for releasing
or stopping the flow of a liquid that flows into or out of the bag. The inlet has a filtering element
for retaining particles possibly produced by the coring phenomenon which can occur when the spike of
the inlet ruptures the plug of the bottle. Also provided is a safety device used for permanently
attaching the bottle to the inlet. '''),
('105', '''An x-ray tube assembly is provided comprising a tube casing assembly including a plurality of
vertical mount posts. An insulator plate is mounted to the plurality of vertical mount posts such that
the insulator plate can translate vertically on the posts. A cathode assembly is mounted to the
insulator plate and generates both an eccentric moment and a vertical expansion in response to a
cathode power load. A semi-compressible element is positioned between at least one of the vertical
mount posts and the insulator plate. The semi-compressible element becomes incompressible at a cathode
power threshold such that the vertical expansion is translated into a correction moment countering the
eccentric moment. ''')
]
cold_df = self.patents_to_df(cold_abstracts)
random_df = self.patents_to_df(random_abstracts)
max_n = 3
min_n = 2
ngram_multiplier = 4
num_ngrams_report = 25
num_ngrams_wordcloud = 25
num_ngrams = max(num_ngrams_report, num_ngrams_wordcloud)
tfidf_cold = TFIDF(cold_df, tokenizer=LemmaTokenizer(), ngram_range=(min_n, max_n))
tfidf_random = TFIDF(random_df, tokenizer=LemmaTokenizer(), ngram_range=(min_n, max_n))
citation_count_dict = {1: 10, 2: 3, 101: 2, 102: 0, 103: 5, 104: 4, 105: 10}
pick = 'sum'
register_writer(TestTableOutput.FakeWriter)
fake_writer = TestTableOutput.FakeWriter('spreadsheet.fake')
table_output(tfidf_cold, tfidf_random, citation_count_dict, num_ngrams, pick, ngram_multiplier, fake_writer)
# Check sheet headings...
self.assertListEqual([None, 'Term', 'Score', 'Rank', 'Focus Score', 'Focus Rank', 'Diff Base to Focus Rank',
'Time Score', 'Time Rank', 'Diff Base to Time Rank', 'Citation Score', 'Citation Rank',
'Diff Base to Citation Rank'], fake_writer.sheets['Summary'][0])
self.assertListEqual([None, 'Term', 'Score', 'Rank'], fake_writer.sheets['Base'][0])
self.assertListEqual([None, 'Term', 'Focus Score', 'Focus Rank'], fake_writer.sheets['Focus'][0])
self.assertListEqual([None, 'Term', 'Time Score', 'Time Rank'], fake_writer.sheets['Time'][0])
self.assertListEqual([None, 'Term', 'Citation Score', 'Citation Rank'], fake_writer.sheets['Cite'][0])
# Base sheet should match summary sheet
for y in range(25):
for x in range(4):
self.assertEqual(fake_writer.sheets['Summary'][y + 1][x], fake_writer.sheets['Base'][y + 1][x])
new_name = archive_existing_file(self.path,
archive_path=None,
tag='creation')
except FileNotFoundError:
pass
else:
self.log(f"archived existing file to {new_name}")
setattr(self, '__file_archived', new_name)
self.log(f"saving")
super().save()
if xlsxwriter is not None:
from pandas.io.excel import register_writer
register_writer(ExcelWriter)
def _make_excel_writer(model, filename, save_now=True, **kwargs):
"""
Write the model to an Excel file.
Parameters
----------
model : larch.Model
filename : str
save_now : bool, default True
Save the model immediately. Set to False if you want to
write additional figures or tables to the file before saving.
**kwargs
Other keyword arguments are passed to the `ExcelWriter`
