Quick Graph Demo in Python

graph.py

class Vertex:
    def __init__(self, data):
        self.data = data

    def __repr__(self):
        return f"{self.data}"


class Graph:
    def __init__(self):
        self.matrix: dict[Vertex, set[Vertex]] = dict()

    def addVertex(self, data):
        v = Vertex(data)
        self.matrix[v] = set()
        return v

    def addEdge(self, u, v):
        self.matrix[u].add(v)
        self.matrix[v].add(u)


g = Graph()

v1 = g.addVertex(4)
v2 = g.addVertex(7)
v3 = g.addVertex(1)
v4 = g.addVertex(3)

g.addEdge(v1, v2)
g.addEdge(v1, v4)

print(g.matrix)