Chore: Fix tests

memgraph · Mar 20, 2024 · 81442f7 · 81442f7
1 parent d5dde27
commit 81442f7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/models/graph.spec.ts b/test/models/graph.spec.ts
@@ -726,17 +726,17 @@ describe('Graph', () => {
       graph.setup({ nodes, edges });
 
       graph.getNodes().forEach((node) => {
-        expect(node.style).toEqual({});
+        expect(node.getStyle()).toEqual({});
       });
 
       graph.getEdges().forEach((edge) => {
-        expect(edge.style).toEqual(edgeStyle);
+        expect(edge.getStyle()).toEqual(edgeStyle);
       });
 
       graph.remove({ edgeIds: graph.getEdges().map((edge) => edge.id) });
 
       graph.getNodes().forEach((node) => {
-        expect(node.style).toEqual(nodeStyle);
+        expect(node.getStyle()).toEqual(nodeStyle);
       });
     });
   });