@@ -46,21 +46,24 @@ def test_load_species_table():
46
46
47
47
# regression test: get_query_payload should work (#308)
48
48
def test_get_payload ():
49
- q = splatalogue .core .Splatalogue .query_lines_async (min_frequency = 1 * u .GHz ,
49
+ payload = splatalogue .core .Splatalogue .query_lines_async (min_frequency = 1 * u .GHz ,
50
50
max_frequency = 10 * u .GHz ,
51
51
get_query_payload = True )
52
- assert q ['body' ]["userInputFrequenciesFrom" ] == [1.0 ]
53
- assert q ['body' ]["userInputFrequenciesTo" ] == [10.0 ]
52
+ payload = json .loads (payload ['body' ])
53
+ assert payload ["userInputFrequenciesFrom" ] == [1.0 ]
54
+ assert payload ["userInputFrequenciesTo" ] == [10.0 ]
54
55
55
56
56
57
# regression test: line lists should ask for only one line list, not all
57
58
def test_line_lists ():
58
- q = splatalogue .core .Splatalogue .query_lines_async (min_frequency = 1 * u .GHz ,
59
+ payload = splatalogue .core .Splatalogue .query_lines_async (min_frequency = 1 * u .GHz ,
59
60
max_frequency = 10 * u .GHz ,
60
61
line_lists = ['JPL' ],
61
62
get_query_payload = True )
62
- assert q ['body' ]['lineListDisplayCDMSJPL' ]
63
- assert not q ['body' ]['lineListDisplaySLAIM' ]
63
+ payload = json .loads (payload ['body' ])
64
+ assert payload ['lineListDisplayJPL' ]
65
+ assert not payload ['lineListDisplaySLAIM' ]
66
+ assert not payload ['lineListDisplayCDMS' ]
64
67
65
68
66
69
# regression test: raise an exception if a string is passed to line_lists
0 commit comments