From 871a4cb0f49eb8854b0b693a3f445a4b4803dc9c Mon Sep 17 00:00:00 2001 From: Leimeroth Date: Tue, 14 Jan 2025 11:48:00 +0100 Subject: [PATCH] add some simple checks --- tests/lammps/test_base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/lammps/test_base.py b/tests/lammps/test_base.py index 9ac619de6..fe8c3d097 100644 --- a/tests/lammps/test_base.py +++ b/tests/lammps/test_base.py @@ -858,10 +858,13 @@ def test_multistep_log_parse(self): file_directory = os.path.join( self.execution_path, "..", "static", "lammps_test_files" ) - _generic_keys_lst, _pressure_dict, _df = _collect_output_log( + _generic_keys_lst, _pressure_dict, df = _collect_output_log( file_name=os.path.join(file_directory, "multi_log_step_test.lammps"), prism=UnfoldingPrism(structure.cell), ) + self.assertEqual(len(df[df.LogStep == 0]), 6) + self.assertEqual(len(df[df.LogStep == 1]), 6) + self.assertEqual(len(df.columns), 6) def collect_dump_file_old(job, file_name="dump.out", cwd=None):