SOF-7354: add snapshot test for colors of atoms with labels

Exabyte-io · May 9, 2024 · 481cb88 · 481cb88
1 parent d165223
commit 481cb88
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 1 deletion.
diff --git a/tests/__tests__/__snapshots__/centos/colorsOfAtomsWithLabels.expected.png b/tests/__tests__/__snapshots__/centos/colorsOfAtomsWithLabels.expected.png
diff --git a/tests/__tests__/__snapshots__/macos/colorsOfAtomsWithLabels.png.png b/tests/__tests__/__snapshots__/macos/colorsOfAtomsWithLabels.png.png
diff --git a/tests/__tests__/__snapshots__/ubuntu/colorsOfAtomsWithLabels.expected.png b/tests/__tests__/__snapshots__/ubuntu/colorsOfAtomsWithLabels.expected.png
diff --git a/tests/__tests__/wave.js b/tests/__tests__/wave.js
@@ -1,4 +1,4 @@
-import { getWaveInstance } from "../enums";
+import { getWaveInstance, getFeOWaveInstance } from "../enums";
 import { takeSnapshotAndAssertEqualityAsync } from "../utils";
 
 test("wave", async () => {
@@ -19,3 +19,14 @@ test("atomRepetitions", async () => {
     wave.rebuildScene();
     return takeSnapshotAndAssertEqualityAsync(wave.renderer.getContext(), "atomRepetitions");
 });
+
+test("colorsOfAtomsWithLabels", async () => {
+    const wave = getFeOWaveInstance();
+    // zoom-out to get full view of the cell
+    wave.toggleOrbitControls();
+    wave.renderer.domElement.dispatchEvent(new WheelEvent("wheel", { deltaY: 1 }));
+    wave.renderer.domElement.dispatchEvent(new WheelEvent("wheel", { deltaY: 1 }));
+    wave.renderer.domElement.dispatchEvent(new WheelEvent("wheel", { deltaY: 1 }));
+    wave.renderer.domElement.dispatchEvent(new WheelEvent("wheel", { deltaY: 1 }));
+    return takeSnapshotAndAssertEqualityAsync(wave.renderer.getContext(), "colorsOfAtomsWithLabels");
+});
diff --git a/tests/enums.js b/tests/enums.js
@@ -19,6 +19,10 @@ export const ELEMENT_PROPERTIES = {
 const materialJsonFilePath = path.resolve(__dirname, "fixtures/material.json");
 export const MATERIAL_CONFIG = JSON.parse(fs.readFileSync(materialJsonFilePath));
 
+// material config for material with atomic labels
+const FeOJsonFilePath = path.resolve(__dirname, "fixtures/FeO.json");
+export const FeO_CONFIG = JSON.parse(fs.readFileSync(FeOJsonFilePath));
+
 export const WAVE_SETTINGS = {
     atomRadiiScale: 0.2,
     repetitions: 1,
@@ -32,3 +36,12 @@ export function getWaveInstance(settings, material = new Made.Material(MATERIAL_
         settings: settings || WAVE_SETTINGS,
     });
 }
+
+export function getFeOWaveInstance(settings, material = new Made.Material(FeO_CONFIG)) {
+    return new Wave({
+        DOMElement: createElement("div", ELEMENT_PROPERTIES),
+        structure: material,
+        cell: material.Lattice.unitCell,
+        settings: settings || WAVE_SETTINGS,
+    });
+}
diff --git a/tests/fixtures/FeO.json b/tests/fixtures/FeO.json