Skip to content

Commit 87f47ff

Browse files
committed
added test of serialiazability
1 parent 61db4fb commit 87f47ff

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

tests/test_bayesian_optimization.py

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
from bayes_opt import BayesianOptimization
55
from bayes_opt.logger import ScreenLogger
66
from bayes_opt.event import Events, DEFAULT_EVENTS
7-
7+
import pickle
8+
import os
89

910
def target_func(**kwargs):
1011
# arbitrary target func
@@ -320,6 +321,21 @@ def test_single_value_objective():
320321
)
321322

322323

324+
def test_pickle():
325+
"""
326+
several users have asked that the BO object be 'pickalable'
327+
This tests that this is the case
328+
"""
329+
optimizer = BayesianOptimization(
330+
f=None,
331+
pbounds={'x': (-10, 10)},
332+
verbose=2,
333+
random_state=1,
334+
)
335+
with open("test_dump.obj", "wb") as filehandler:
336+
pickle.dump(optimizer, filehandler)
337+
os.remove('test_dump.obj')
338+
323339
if __name__ == '__main__':
324340
r"""
325341
CommandLine:

0 commit comments

Comments
 (0)