From 3aba22d6501785da6c285dc9e32e61bf7cc32447 Mon Sep 17 00:00:00 2001 From: David H Hagan Date: Fri, 8 Jul 2022 12:23:52 -0400 Subject: [PATCH] Support for new unittests --- tests/test_concat.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_concat.py b/tests/test_concat.py index 01562d3..228b777 100644 --- a/tests/test_concat.py +++ b/tests/test_concat.py @@ -46,10 +46,14 @@ def test_concat_files_csv(self): # are the number of lines correct? df1 = pd.read_csv(os.path.join(self.test_files_dir, "lcs-1.csv")) df2 = pd.read_csv(os.path.join(self.test_files_dir, "lcs-2.csv"), skiprows=1) - df3 = pd.read_csv(os.path.join(self.test_dir, "output.csv")) + df3 = pd.read_csv(os.path.join(self.test_dir, "output.csv")).sort_values('timestamp') self.assertEqual(df1.shape[0] + df2.shape[0], df3.shape[0]) + # Are the timestamps correct? + for i, (_, r) in enumerate(df2.sort_values('timestamp').head().iterrows()): + self.assertEqual(r['timestamp'], df3.loc[i, "timestamp"]) + def test_concat_logfiles(self): runner = CliRunner() result = runner.invoke(concat,