lattice_builder modified

pyMBE-dev · Dec 1, 2024 · 3e84660 · 3e84660
1 parent 76c384f
commit 3e84660
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/testsuite/lattice_builder.py b/testsuite/lattice_builder.py
@@ -160,8 +160,8 @@ def test_lattice_setup(self):
                          pmb.set_chain,
                          "[0 0 0]", "[0 0 0]", node_positions, espresso_system=espresso_system)
 
-        lattice.set_colormap(lattice.colormap)
-        for index, (label, color) in enumerate(lattice.colormap.items()):
+        lattice.set_colormap(self.colormap)
+        for index, (label, color) in enumerate(self.colormap.items()):
             np.testing.assert_equal(actual = lattice.get_monomer_color(label),desired = color, verbose=True)
             np.testing.assert_equal(actual = lattice.get_monomer_color_index(label),desired = index, verbose=True)