def test_accelerator_experiment_many(get_records_mock):
schema = AcceleratorExperimentSchemaV1()
dump1 = {
"legacy_name": "LEGACY-EXP1",
"institutions": [{
"value": "INS"
}],
"accelerator": {
"value": "ACC"
},
"experiment": {
"value": "EXP1"
},
}
dump2 = {
"legacy_name": "LEGACY-EXP2",
"institutions": [{
"value": "INS"
}],
"accelerator": {
"value": "ACC"
},
"experiment": {
"value": "EXP2"
},
}
expected = [{"name": "INS-ACC-EXP1"}, {"name": "INS-ACC-EXP2"}]
result = schema.dumps([dump1, dump2], many=True).data
assert expected == orjson.loads(result)
def test_returns_none_as_name_if_empty_present(get_records_mock):
schema = AcceleratorExperimentSchemaV1()
dump = {}
expected = {"name": None}
record = LiteratureRecord(dump)
result = schema.dumps(record).data
assert expected == orjson.loads(result)
def test_returns_legacy_name_as_name(get_records_mock):
schema = AcceleratorExperimentSchemaV1()
dump = {"legacy_name": "Test"}
record = LiteratureRecord(dump)
expected = {"name": "Test"}
result = schema.dumps(record).data
assert expected == orjson.loads(result)
def test_returns_legacy_name_as_name_if_experiment_missing(get_records_mock):
schema = AcceleratorExperimentSchemaV1()
dump = {
"legacy_name": "LEGACY-EXP1",
"institutions": [{
"value": "INS"
}],
"accelerator": {
"value": "ACC"
},
}
expected = {"name": "LEGACY-EXP1"}
record = LiteratureRecord(dump)
result = schema.dumps(record).data
assert expected == orjson.loads(result)
def test_returns_dashed_institution_accelerator_experiment_as_name_with_unicode(
get_records_mock, ):
schema = AcceleratorExperimentSchemaV1()
dump = {
"legacy_name": "LEGACY-EXP1",
"institutions": [{
"value": "PSI, Villigen"
}],
"accelerator": {
"value": "PSI πM1 beam line"
},
"experiment": {
"value": "MUSE"
},
}
expected = {"name": "PSI, Villigen-PSI πM1 beam line-MUSE"}
record = LiteratureRecord(dump)
result = schema.dumps(record).data
assert expected == orjson.loads(result)
def test_returns_dashed_institution_accelerator_experiment_as_name_if_all_present(
get_records_mock):
schema = AcceleratorExperimentSchemaV1()
dump = {
"legacy_name": "LEGACY-EXP1",
"institutions": [{
"value": "INS"
}],
"accelerator": {
"value": "ACC"
},
"experiment": {
"value": "EXP1"
},
}
expected = {"name": "INS-ACC-EXP1"}
record = LiteratureRecord(dump)
result = schema.dumps(record).data
assert expected == json.loads(result)