diff --git a/test/query_test.py b/test/query_test.py index f36300ab..17d318df 100644 --- a/test/query_test.py +++ b/test/query_test.py @@ -308,20 +308,20 @@ def test_get_references(query): ref = query.parse_ref('ksm+06') assert isinstance(ref, string_types) - assert "Tests of General Relativity from Timing the Double Pulsar" in ref + assert "2006, Science, 314, 97" in ref # test parsing two references ref = query.parse_ref(['ksm+06', 'abb+18']) assert len(ref) == 2 - assert "Tests of General Relativity from Timing the Double Pulsar" in ref[0] - assert "The NANOGrav 11-year Data Set" in ref[1] + assert "2006, Science, 314, 97" in ref[0] + assert "2018, ApJS, 235, 37" in ref[1] # test parsing two references (with the second one being gibberish) ref = query.parse_ref(['ksm+06', 'wlihlacljkblf']) assert len(ref) == 2 - assert "Tests of General Relativity from Timing the Double Pulsar" in ref[0] + assert "2006, Science, 314, 97" in ref[0] assert ref[1] is None