test sample() only when main
This commit is contained in:
@@ -75,12 +75,14 @@ def to_state_string(state, state_nex=None):
|
||||
|
||||
return state_str
|
||||
|
||||
print("Testing")
|
||||
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])
|
||||
test_sample([0.1, 0.1])
|
||||
test_sample([0.3, 0.0, 0.2, 0.0])
|
||||
test_sample([0.0, 0.0, 0.0, 0.0])
|
||||
test_sample([0.5, 0.0, 0.2, 0.3])
|
||||
test_sample([0.3, 0.1, 0.4, 0.2])
|
||||
|
||||
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])
|
||||
test_sample([0.1, 0.1])
|
||||
test_sample([0.3, 0.0, 0.2, 0.0])
|
||||
test_sample([0.0, 0.0, 0.0, 0.0])
|
||||
test_sample([0.5, 0.0, 0.2, 0.3])
|
||||
test_sample([0.3, 0.1, 0.4, 0.2])
|
||||
|
||||
Reference in New Issue
Block a user