Skip to content

Commit cd9bcd9

Browse files
committed
fix issues
1 parent fc56375 commit cd9bcd9

File tree

1 file changed

+26
-1
lines changed

1 file changed

+26
-1
lines changed

_unittests/ut_helpers/test_log_helper.py

Lines changed: 26 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def test_enumerate_csv_files(self):
204204
self.assertEqual((3, 11), cube.shape)
205205
self.assertIn("RAWFILENAME", cube.data.columns)
206206

207-
def test_cube_logs_performance(self):
207+
def test_cube_logs_performance1(self):
208208
output = self.get_dump_file("test_cube_logs_performance.xlsx")
209209
filename = os.path.join(os.path.dirname(__file__), "data", "data-agg.zip")
210210
assert list(enumerate_csv_files(filename))
@@ -229,6 +229,31 @@ def test_cube_logs_performance(self):
229229
)
230230
self.assertExists(output)
231231

232+
def test_cube_logs_performance2(self):
233+
output = self.get_dump_file("test_cube_logs_performance.xlsx")
234+
filename = os.path.join(os.path.dirname(__file__), "data", "data-agg.zip")
235+
assert list(enumerate_csv_files(filename))
236+
dfs = [open_dataframe(df) for df in enumerate_csv_files(filename)]
237+
assert dfs, f"{filename!r} empty"
238+
cube = CubeLogsPerformance(dfs, keep_last_date=True)
239+
cube.load()
240+
cube.to_excel(
241+
output,
242+
views=[
243+
"agg-suite",
244+
"disc",
245+
"speedup",
246+
"counts",
247+
"time",
248+
"time_export",
249+
"err",
250+
# "cmd",
251+
"bucket-speedup",
252+
"raw-short",
253+
],
254+
)
255+
self.assertExists(output)
256+
232257
def test_duplicate(self):
233258
df = pandas.DataFrame(
234259
[

0 commit comments

Comments
 (0)