vary the public session checking
@@ -17,6 +17,7 @@ def test_networkml_eval_onelayer():
def test_networkml_eval_randomforest():
sys.argv = ['bin/networkml', '-p', 'tests/', '-a', 'randomforest']
+ os.environ['POSEIDON_PUBLIC_SESSIONS'] = 0
netml = NetworkML()
@@ -36,6 +37,7 @@ def test_networkml_train_onelayer():
def test_networkml_train_randomforest():
sys.argv = ['bin/networkml', '-p', 'tests/',
'-o', 'train', '-a', 'randomforest', '-m', 'networkml/trained_models/randomforest/RandomForestModel.pkl']
+ os.environ['POSEIDON_PUBLIC_SESSIONS'] = 0
with pytest.raises(SystemExit) as pytest_wrapped_e:
netml = NetworkML()
assert pytest_wrapped_e.type == SystemExit
@@ -59,4 +61,5 @@ def test_networkml_test_onelayer():
def test_networkml_test_randomforest():
sys.argv = ['bin/networkml', '-p', 'tests/',
'-o', 'test', '-a', 'randomforest', '-m', 'networkml/trained_models/randomforest/RandomForestModel.pkl']
+ os.environ['POSEIDON_PUBLIC_SESSIONS'] = 0
netml = NetworkML()
