1# Add path to source code 2import sys, os 3if os.getcwd().endswith('tests'): 4 sys.path.insert(0, '..') 5elif os.getcwd().endswith('websocket-server'): 6 sys.path.insert(0, '.') 7