@@ -72,16 +72,18 @@ def test_seed_generator():
7272 """
7373 k_list = [1 , 2 ]
7474 replicates = [1 , 2 , 3 ]
75- extra_options = ""
75+ seed = 1235813
7676
7777 mock_jobs = [(2 , 3 ), (2 , 2 ), (2 , 1 ), (1 , 3 ), (1 , 2 ), (1 , 1 )]
78- returned_jobs = sw .seed_generator (extra_options , k_list , replicates )
78+ mock_jobs = [("2153978" , 2 , 3 ), ("940261" , 2 , 2 ), ("8867621" , 2 , 1 ),
79+ ("786598" , 1 , 3 ), ("3922463" , 1 , 2 ), ("6870574" , 1 , 1 )]
80+ returned_jobs = sw .seed_generator (seed , k_list , replicates )
7981 assert returned_jobs == mock_jobs
8082
81- # Add a seed option
82- extra_options = "-D 1234"
83+ # Use a different seed
84+ seed = 42
8385
84- mock_jobs = [("7394266 " , 2 , 3 ), ("1960426 " , 2 , 2 ), ("125686 " , 2 , 1 ),
85- ("1520747 " , 1 , 3 ), ("9768157 " , 1 , 2 ), ("585963 " , 1 , 1 )]
86- returned_jobs = sw .seed_generator (extra_options , k_list , replicates )
86+ mock_jobs = [("1867825 " , 2 , 3 ), ("419610 " , 2 , 2 ), ("4614226 " , 2 , 1 ),
87+ ("4108603 " , 1 , 3 ), ("3744854 " , 1 , 2 ), ("2341057 " , 1 , 1 )]
88+ returned_jobs = sw .seed_generator (seed , k_list , replicates )
8789 assert returned_jobs == mock_jobs
0 commit comments