test sample() only when main

2024-06-09 17:30:24 +02:00
parent 7cad1442df
commit 19c972bc7d
1 changed files with 11 additions and 9 deletions
@@ -75,7 +75,9 @@ def to_state_string(state, state_nex=None):

    return state_str

-print("Testing")
+
+if __name__ == '__main__':
+    print("Testing sample()")
    test_sample([0.7, 0.1, 0.1, 0.1])
    test_sample([0.1, 0.1, 0.2, 0.1])
    test_sample([0.2, 0.8])