fix: adjacency matrix

Add no_edge_value in copy, float conversion in ad_edge
rubenperezm · Oct 7, 2024 · d701651 · d701651
1 parent 0c6c39d
commit d701651
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pystrukts/graph/adjacency_matrix.py b/pystrukts/graph/adjacency_matrix.py
@@ -79,7 +79,8 @@ def copy(self) -> 'AdjacencyMatrix':
             out (AdjacencyMatrix): The copy of the graph.
         '''
 
-        return AdjacencyMatrix(directed = self.directed, matrix = self.matrix)
+        return AdjacencyMatrix(directed = self.directed,
+                               matrix = self.matrix, no_edge_value = self.no_edge_value)
 
     def _copy_matrix(self, matrix: list, directed: bool) -> list:
         '''
@@ -170,10 +171,10 @@ def add_edge(self, x: int, y: int, w: float = 1.0) -> None:
         self._raise_error_if_invalid_vertex(x)
         self._raise_error_if_invalid_vertex(y)
 
-        self.matrix[x][y] = w
+        self.matrix[x][y] = float(w)
 
         if self.directed:
-            self.matrix[y][x] = w
+            self.matrix[y][x] = float(w)
 
     def remove_edge(self, x: int, y: int) -> None:
         '''

diff --git a/tests/test_adjacency_matrix.py b/tests/test_adjacency_matrix.py
@@ -48,6 +48,7 @@ def test_copy():
     assert graph2.num_vertices == 3
     assert graph2.directed == False
     assert graph2.matrix == matrix
+    assert graph2.no_edge_value == 0
 
 def test_has_edge():
     matrix = [[0, 1, 0], [0, 0, 0], [1, 0, 1]]