diff --git a/graphs/tests/test_graphs_floyd_warshall.py b/graphs/tests/test_graphs_floyd_warshall.py
index f87843112..aae312aac 100644
--- a/graphs/tests/test_graphs_floyd_warshall.py
+++ b/graphs/tests/test_graphs_floyd_warshall.py
@@ -18,7 +18,7 @@ def test_no_edges():
     assert dist == expected
 
 
-def test_example_input():
+def test_example_input(capsys):
     num_vertices = 3
     graph = [
         [float("inf"), float("inf"), float("inf")],
@@ -35,13 +35,10 @@ def test_example_input():
         [float("inf"), float("inf"), 0],
     ]
     dist, _ = floyd_warshall(graph, num_vertices)
+    _ = capsys.readouterr()
     assert dist == expected
 
 
-if __name__ == "__main__":
-    pytest.main()
-
-
 def test_unreachable_vertices():
     graph = [
         [0, 1, float("inf")],