From 885f7c3f9915adada0d2ef62e679d60e4ac335e7 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Aug 2025 14:58:56 +0200
Subject: [PATCH] chore: fixing graph elements tests

---
 cognee/modules/graph/cognee_graph/CogneeGraphElements.py  | 2 +-
 .../unit/modules/graph/cognee_graph_elements_test.py      | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/cognee/modules/graph/cognee_graph/CogneeGraphElements.py b/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
index c22cc5c18..2d8917d4a 100644
--- a/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
+++ b/cognee/modules/graph/cognee_graph/CogneeGraphElements.py
@@ -106,7 +106,7 @@ class Edge:
         dimension: int = 1,
     ):
         if dimension <= 0:
-            InvalidDimensionsError()
+            raise InvalidDimensionsError()
         self.node1 = node1
         self.node2 = node2
         self.attributes = attributes if attributes is not None else {}
diff --git a/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py b/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
index a852bcee3..a8fb17268 100644
--- a/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
+++ b/cognee/tests/unit/modules/graph/cognee_graph_elements_test.py
@@ -15,7 +15,7 @@ def test_node_initialization():
 
 def test_node_invalid_dimension():
     """Test that initializing a Node with a non-positive dimension raises an error."""
-    with pytest.raises(InvalidDimensionsError, match="Dimensions must be a positive integers"):
+    with pytest.raises(InvalidDimensionsError):
         Node("node1", dimension=0)
 
 
@@ -68,7 +68,7 @@ def test_is_node_alive_in_dimension():
 def test_node_alive_invalid_dimension():
     """Test that checking alive status with an invalid dimension raises an error."""
     node = Node("node1", dimension=1)
-    with pytest.raises(DimensionOutOfRangeError, match="Dimension 1 is out of range"):
+    with pytest.raises(DimensionOutOfRangeError):
         node.is_node_alive_in_dimension(1)
 
 
@@ -105,7 +105,7 @@ def test_edge_invalid_dimension():
     """Test that initializing an Edge with a non-positive dimension raises an error."""
     node1 = Node("node1")
     node2 = Node("node2")
-    with pytest.raises(DimensionOutOfRangeError, match="Dimensions must be a positive integer."):
+    with pytest.raises(InvalidDimensionsError):
         Edge(node1, node2, dimension=0)
 
 
@@ -124,7 +124,7 @@ def test_edge_alive_invalid_dimension():
     node1 = Node("node1")
     node2 = Node("node2")
     edge = Edge(node1, node2, dimension=1)
-    with pytest.raises(DimensionOutOfRangeError, match="Dimension 1 is out of range"):
+    with pytest.raises(DimensionOutOfRangeError):
         edge.is_edge_alive_in_dimension(1)