19
19
"input_uri" : "helloworld.ipynb" ,
20
20
"runtime_environment_name" : "default" ,
21
21
"name" : "hello world" ,
22
- "output_formats" : ["ipynb" ]
22
+ "output_formats" : ["ipynb" ],
23
23
}
24
24
25
+
25
26
def test_create_job_definition (jp_scheduler ):
26
27
with patch ("jupyter_scheduler.scheduler.fsspec" ) as mock_fsspec :
27
28
with patch ("jupyter_scheduler.scheduler.Scheduler.file_exists" ) as mock_file_exists :
28
29
mock_file_exists .return_value = True
29
- job_definition_id = jp_scheduler .create_job_definition (CreateJobDefinition (** test_job_definition_parameters ))
30
+ job_definition_id = jp_scheduler .create_job_definition (
31
+ CreateJobDefinition (** test_job_definition_parameters )
32
+ )
30
33
31
34
with jp_scheduler .db_session () as session :
32
35
definitions = session .query (JobDefinition ).all ()
@@ -40,10 +43,8 @@ def test_create_job_definition(jp_scheduler):
40
43
assert [] == definition .on_events
41
44
42
45
43
- event_type_parameters = {
44
- "name" : 'type1' ,
45
- "parameters" : {'param1' : 'value1' }
46
- }
46
+ event_type_parameters = {"name" : "type1" , "parameters" : {"param1" : "value1" }}
47
+
47
48
48
49
def test_create_job_definition_with_on_events (jp_scheduler ):
49
50
with patch ("jupyter_scheduler.scheduler.fsspec" ) as mock_fsspec :
@@ -57,7 +58,7 @@ def test_create_job_definition_with_on_events(jp_scheduler):
57
58
definitions = session .query (JobDefinition ).all ()
58
59
definition = definitions [0 ]
59
60
assert 1 == len (definition .on_events )
60
- assert [{' name' : ' type1' , ' parameters' : {' param1' : ' value1' }}] == definition .on_events
61
+ assert [{" name" : " type1" , " parameters" : {" param1" : " value1" }}] == definition .on_events
61
62
62
63
63
64
job_definition_1 = {
0 commit comments