test sample() only when main

This commit is contained in:
2024-06-09 17:30:24 +02:00
parent 7cad1442df
commit 19c972bc7d
+3 -1
View File
@@ -75,7 +75,9 @@ def to_state_string(state, state_nex=None):
return state_str return state_str
print("Testing")
if __name__ == '__main__':
print("Testing sample()")
test_sample([0.7, 0.1, 0.1, 0.1]) test_sample([0.7, 0.1, 0.1, 0.1])
test_sample([0.1, 0.1, 0.2, 0.1]) test_sample([0.1, 0.1, 0.2, 0.1])
test_sample([0.2, 0.8]) test_sample([0.2, 0.8])