From 217fbae427ffc0bf7758c6776c077900b671bfb8 Mon Sep 17 00:00:00 2001
From: Pranab Das <31024886+pranabdas@users.noreply.github.com>
Date: Wed, 8 May 2024 19:47:25 +0800
Subject: [PATCH] SOF-7354: update atomName in tests

---
 tests/__tests__/mixins/labels.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/__tests__/mixins/labels.js b/tests/__tests__/mixins/labels.js
index 0a7a81f..3c3b92c 100644
--- 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,