ci: Add tests for LHEEventInfo, LHEParticle, LHEFile classes

src/pylhe/__init__.py

@@ -16,7 +16,6 @@
     "LHEInit",
     "LHEParticle",
     "LHEProcInfo",
-    "loads",
     "read_lhe",
     "read_lhe_init",
     "read_lhe_with_attributes",
@@ -122,7 +121,9 @@ class LHEEventInfo:
 
     def __init__(self, **kwargs):
         if set(kwargs.keys()) != set(self.fieldnames):
-            raise RuntimeError
+            raise RuntimeError(
+                f"LHEEventInfo constructor expects fields {self.fieldnames}! Got {kwargs.keys()}."
+            )
         for k, v in kwargs.items():
             setattr(self, k, v)
 
@@ -150,7 +151,9 @@ class LHEParticle:
 
     def __init__(self, **kwargs):
         if set(kwargs.keys()) != set(self.fieldnames):
-            raise RuntimeError
+            raise RuntimeError(
+                f"LHEParticle constructor expects fields {self.fieldnames}! Got {kwargs.keys()}."
+            )
         for k, v in kwargs.items():
             setattr(self, k, v)
 
@@ -203,10 +206,6 @@ def fromstring(cls, string):
         return dict(zip(cls.fieldnames, map(float, string.split())))
 
 
-def loads():
-    pass
-
-
 def _extract_fileobj(filepath):
     """
     Checks to see if a file is compressed, and if so, extract it with gzip

tests/test_api.py

@@ -19,7 +19,6 @@ def test_top_level_api():
         "LHEParticle",
         "LHEProcInfo",
         "__version__",
-        "loads",
         "read_lhe",
         "read_lhe_init",
         "read_lhe_with_attributes",

tests/test_classes.py

@@ -0,0 +1,42 @@
+import pytest
+import skhep_testdata
+
+from pylhe import LHEEventInfo, LHEFile, LHEInit, LHEParticle, LHEProcInfo, read_lhe
+
+TEST_FILE = skhep_testdata.data_path("pylhe-testfile-pr29.lhe")
+
+
+def test_LHEEvent():
+    events = read_lhe(TEST_FILE)
+    event = next(events)  # it contains 8 pions and a proton
+
+    assert event.eventinfo is not None
+
+    assert len(event.particles) == 9
+
+    for p in event.particles:
+        assert p.event == event
+
+    assert event._graph is None
+
+
+def test_LHEEventInfo_no_default_init():
+    with pytest.raises(RuntimeError):
+        _ = LHEEventInfo()
+
+
+def test_LHEFile_default_init():
+    assert LHEFile() is not None
+
+
+def test_LHEInit_default_init():
+    assert LHEInit() is not None
+
+
+def test_LHEParticle_no_default_init():
+    with pytest.raises(RuntimeError):
+        _ = LHEParticle()
+
+
+def test_LHEProcInfo_default_init():
+    assert LHEProcInfo() is not None