test: add test for graph feature

test/test_integration.py

 import sys
 
 from conftest import *
-sys.path.append('../user_session_decompo')
-from swf2sessions import swf2sessions
 
 import os, subprocess
 import pytest
@@ -18,11 +16,14 @@ def run_test(delim,
     out_dir = os.path.abspath(f'test-out/{test_name}')
     create_dir_rec_if_needed(out_dir)
 
-    cp = subprocess.run(
-        ['python3', 'swf2sessions.py', '-at', str(threshold), input_swf, out_dir],
-        check=True)
-
-    swf2sessions(input_swf, out_dir, delim, threshold, dyn_red, graph, False)
+    if not graph:
+        cp = subprocess.run(['python3', 'swf2sessions.py', 
+            '-at', str(threshold), '-q', input_swf, out_dir],
+            check=True)
+    else:
+        cp = subprocess.run(['python3', 'swf2sessions.py', 
+            '-at', str(threshold), '--graph', '-q', input_swf, out_dir],
+            check=True)
 
     return out_dir, cp
 
@@ -72,6 +73,17 @@ def test_kth_delim_max():
         assert len(os.listdir(out_dir)) > 0
 
 
+def test_graph():
+    """Launch a couple of instances with the --graph option and checks if the graphs are output"""
+
+    instances = ["", "", ""]
+    instances[0], _ = run_test('arrival', 0, graph=True)
+    instances[1], _ = run_test('last', 30, graph=True)
+    instances[2], _ = run_test('max', 1440, graph=True)
 
+    for out_dir in instances:
+        assert os.path.exists(out_dir)
+        assert os.path.exists(f"{out_dir}/graphs")
+        assert len(os.listdir(f"{out_dir}/graphs")) == KTH_WL.nb_users