def test_parse_raw_schema(): schema = TaskMetadata( id="test_id", deviceId="device_id", shots=1000, ) assert schema == BraketSchemaBase.parse_raw_schema(schema.json()) assert isinstance(schema, TaskMetadata)
def test_import_schema_module(): schema = TaskMetadata( id="test_id", deviceId="device_id", shots=1000, ) module = BraketSchemaBase.import_schema_module(schema) assert schema == module.TaskMetadata.parse_raw(schema.json())
def test_correct_metadata_all(id, device_id, shots): device_parameters = { "braketSchemaHeader": { "name": "braket.device_schema.rigetti.rigetti_device_parameters", "version": "1", }, "paradigmParameters": { "braketSchemaHeader": { "name": "braket.device_schema.gate_model_parameters", "version": "1", }, "qubitCount": 1, }, } createdAt = "2020-01-02T03:04:05.006Z" endedAt = "2020-01-03T03:04:05.006Z" status = "COMPLETED" failureReason = "Foo" metadata = TaskMetadata( id=id, deviceId=device_id, shots=shots, deviceParameters=device_parameters, createdAt=createdAt, endedAt=endedAt, status=status, failureReason=failureReason, ) assert metadata.id == id assert metadata.deviceId == device_id assert metadata.shots == shots assert metadata.deviceParameters == device_parameters assert metadata.createdAt == createdAt assert metadata.endedAt == endedAt assert metadata.status == status assert metadata.failureReason == failureReason assert TaskMetadata.parse_raw(metadata.json()) == metadata assert metadata == TaskMetadata.parse_raw_schema(metadata.json())
def task_metadata(id, device_id, shots): return TaskMetadata(id=id, deviceId=device_id, shots=shots)
def test_incorrect_shots(id, device_id, shots): TaskMetadata(id=id, deviceId=device_id, shots=shots)
def test_correct_metadata_minimum(id, device_id, shots): metadata = TaskMetadata(id=id, deviceId=device_id, shots=shots) assert metadata.id == id assert metadata.deviceId == device_id assert metadata.shots == shots assert TaskMetadata.parse_raw(metadata.json()) == metadata
def test_incorrect_header(braket_schema_header, id, device_id, shots): TaskMetadata(braketSchemaHeader=braket_schema_header, id=id, deviceId=device_id, shots=shots)
def test_missing_properties(): TaskMetadata()