test.py

from env import ThreatChainEnv

env = ThreatChainEnv()

state = env.reset()
print(state)

state, reward, done, _ = env.step("phishing_attack")

print(state)
print(reward, done)