fix tests

Author: janosh

src/pymatgen/io/lammps/data.py

@@ -1148,8 +1148,8 @@ def find_eq_types(label, section) -> list:
         for dct in topo_coeffs[kw]:
             main_data.append(dct["coeffs"])
             distinct_types.append(dct["types"])
-            for key, val in class2_data.items():
-                class2_data[key].append(val)
+            for key, lst in class2_data.items():
+                lst.append(dct[key])
         distinct_types = [set(itertools.chain(*(find_eq_types(t, kw) for t in dt))) for dt in distinct_types]
         type_counts = sum(len(dt) for dt in distinct_types)
         type_union = set.union(*distinct_types)

tests/io/vasp/test_outputs.py

@@ -1830,7 +1830,7 @@ def test_n2_spin(self):
 
         orig_gen_g_points = Wavecar._generate_G_points
         try:
-            Wavecar._generate_G_points = lambda _x, _y, _gamma: []
+            Wavecar._generate_G_points = lambda _x, _y, gamma: []
             with pytest.raises(ValueError, match=r"not enough values to unpack \(expected 3, got 0\)"):
                 Wavecar(f"{VASP_OUT_DIR}/WAVECAR.N2")
         finally: