SOF-7354: update atomName in tests

Exabyte-io · May 8, 2024 · 217fbae · 217fbae
1 parent 64533a3
commit 217fbae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/__tests__/mixins/labels.js b/tests/__tests__/mixins/labels.js
@@ -19,7 +19,7 @@ describe("Atom labels", () => {
         const basisAtomsNumber = wave.structure.basis.elements.length;
 
         atoms.forEach((atom) => {
-            const atomName = atom.name.split("-")[0];
+            const atomName = atom.nameWithLabel;
             const atomPosition = new THREE.Vector3().setFromMatrixPosition(atom.matrixWorld);
 
             const offsetVector = wave.getLabelOffsetVector(atomPosition, atomName);
@@ -49,7 +49,7 @@ describe("Atom labels", () => {
 
         const basisAtomsNumber = wave.structure.basis.elements.length;
         const doAllAtomsHaveLabels = atoms.every((atom) => {
-            const atomName = atom.name.split("-")[0];
+            const atomName = atom.nameWithLabel;
             const labelName = `labels-for-${atomName}`;
             const labelPointsByAtomName = labels.find(
                 (labelPoints) => labelPoints.name === labelName,