def test_set_subschema2():
# Scenario values:
test_value_1 = {
'record_Type': 'schema:test',
'RECORD_ID': 'test123',
'credibility': 50,
'schema:name': 'test_name',
'schema:test': 'test_var',
'schema:sub_schema': {
'@type': 'schema:sub',
'@id': 'sub_id',
'schema:sub_var': 'testsubvar'
}
}
expected_result = {'@type': 'schema:test', '@id': 'test123', 'schema:name': [{'value': 'test_name', 'credibility': 50}], 'schema:test': [{'value': 'test_var', 'credibility': 50}], 'schema:sub_schema': [{'value': {'@type': 'schema:sub', '@id': 'sub_id'}, 'credibility': 50}]}
# Test conditions
kr = Kraken_record()
kr.set(test_value_1)
actual_result = kr.get()
# Test evaluation
assert(actual_result == expected_result)
def test_get_simple():
# Scenario values:
test_value = {
'record_Type': 'schema:test',
'RECORD_ID': 'test123',
'credibility': 50,
'schema:name': 'test_name',
'schema:test': 'test_var'
}
expected_result = {'@type': 'schema:test', '@id': 'test123', 'schema:name': 'test_name', 'schema:test': 'test_var'}
# Test conditions
kr = Kraken_record()
kr.set(test_value)
actual_result = kr.get_simple()
# Test evaluation
assert(actual_result == expected_result)
def test_clean_input_value_complex():
# Scenario values:
test_value = {
'record_Type': 'schema:test',
'RECORD_ID': 'test123',
'kraken:name': 'test_name'
}
expected_result = {
'kraken:record_type': 'schema:test',
'kraken:record_id': 'test123',
'kraken:name': 'test_name'
}
# Test conditions
kr = Kraken_record()
kr.input_value = test_value
kr._clean_input_value_keys()
actual_result = kr.input_value
# Test evaluation
assert(actual_result == expected_result)
def test_set_no_id():
# Scenario values:
test_value = {
'record_Type': 'schema:webpage',
'credibility': 50,
'schema:name': 'test_name',
'schema:test': 'test_var',
'schema:url': 'https://www.test.com'
}
expected_result = {'@type': 'schema:webpage', '@id': 'https://www.test.com', 'schema:name': [{'value': 'test_name', 'credibility': 50}], 'schema:test': [{'value': 'test_var', 'credibility': 50}], 'schema:url': [{'value': 'https://www.test.com', 'credibility': 50}]}
# Test conditions
kr = Kraken_record()
kr.set(test_value)
actual_result = kr.get()
# Test evaluation
assert(actual_result == expected_result)
def test_equal_true():
# Scenario values:
test_value_1 = {
'record_Type': 'schema:test',
'RECORD_ID': 'test123',
'credibility': 50,
'schema:name': 'test_name',
'schema:test': 'test_var'
}
expected_result = True
# Test conditions
kr1 = Kraken_record()
kr1.set(test_value_1)
kr2 = Kraken_record()
kr2.set(test_value_1)
actual_result = kr1 == kr2
# Test evaluation
assert(actual_result == expected_result)
def test_add_same_values():
# Scenario values:
test_value_1 = {
'record_Type': 'schema:test',
'RECORD_ID': 'test123',
'credibility': 50,
'schema:name': 'test_name',
'schema:test': 'test_var'
}
expected_result = {'@type': 'schema:test', '@id': 'test123', 'schema:name': [{'value': 'test_name', 'credibility': 50}], 'schema:test': [{'value': 'test_var', 'credibility': 50}]}
# Test conditions
kr1 = Kraken_record()
kr1.set(test_value_1)
kr2 = Kraken_record()
kr2.set(test_value_1)
kr_total = kr1 + kr2
actual_result = kr_total.get()
# Test evaluation
assert(actual_result == expected_result)
def test_set_attribute():
# Scenario values:
test_value = {
'record_Type': 'schema:test',
'RECORD_ID': 'test123',
'kraken:name': 'test_name'
}
expected_result = {
'kraken:record_type': 'schema:test',
'kraken:record_id': 'test123',
'kraken:name': 'test_name'
}
attr_value = {'value': 'test_2', 'credibility': 40}
expected_result = [{'value': 'test_2', 'credibility': 40}]
# Test conditions
kr = Kraken_record()
kr.record_type = 'schema:test'
kr.record_id = 'test123'
kr.set_attr('schema:test_attr', attr_value)
actual_result = kr.get_attr_record('schema:test_attr')
# Test evaluation
assert(actual_result == expected_result)
def test_set_attribute_multiple():
# Scenario values:
attr_value1 = {'value': 'test_1', 'kraken:credibility': 30}
attr_value2 = {'value': 'test_2', 'kraken:credibility': 40}
expected_result = [{'value': 'test_2', 'credibility': 40}, {'value': 'test_1', 'credibility': 30}]
# Test conditions
kr = Kraken_record()
kr.record_type = 'schema:test'
kr.record_id = 'test123'
kr.set_attr('schema:test_attr', attr_value1)
kr.set_attr('schema:test_attr', attr_value2)
actual_result = kr.get_attr_record('schema:test_attr')
# Test evaluation
assert(actual_result == expected_result)
def test_clean_input_value_blank():
# Scenario values:
kr = Kraken_record()
test_value = None
# Test conditions
kr.input_value = test_value
kr._clean_input_value_keys()
# Expected outcome
result_value = None
# Test evaluation
assert(kr.input_value == result_value)
def test_dump():
# Scenario values:
test_value = {'@type': 'schema:test', '@id': 'test123', 'schema:name': [{'value': 'test_name', 'credibility': 50}], 'schema:test': [{'value': 'test_var', 'credibility': 50}]}
expected_result = {'kraken:record_type': 'schema:test', 'kraken:record_id': 'test123', 'schema:name': [{'value': [{'value': 'test_name', 'credibility': 50}]}], 'schema:test': [{'value': [{'value': 'test_var', 'credibility': 50}]}]}
# Test conditions
kr = Kraken_record()
kr.set(test_value)
actual_result = kr.dump()
# Test evaluation
assert(actual_result == expected_result)
def test_set_2():
# Scenario values:
test_value_1 = {
'record_Type': 'schema:test',
'RECORD_ID': 'test123',
'credibility': 50,
'schema:name': 'test_name',
'schema:test': 'test_var'
}
test_value_2 = {
'record_Type': 'schema:test',
'RECORD_ID': 'test123',
'credibility': 40,
'schema:name': 'test_name2',
'schema:test': 'test_var2'
}
expected_result = {'@type': 'schema:test', '@id': 'test123', 'schema:name': [{'value': 'test_name', 'credibility': 50}, {'value': 'test_name2', 'credibility': 40}], 'schema:test': [{'value': 'test_var', 'credibility': 50}, {'value': 'test_var2', 'credibility': 40}]}
# Test conditions
kr = Kraken_record()
kr.set(test_value_1)
kr.set(test_value_2)
actual_result = kr.get()
# Test evaluation
assert(actual_result == expected_result)
import os
test = False
if test:
os.system("pip install pytest")
os.system("python -m pytest tests")
record = {
"@type": "schema:person",
"@id": "test_id",
"schema:givenname": "test Name",
"schema:familyname": "Test last name ",
"schema:email": " ",
"schema:telephone": " ",
"schema:hasOccupation": {
"@type": "schema:Occupation",
"schema:name": "some occupation"
},
"schema:worksfor": {
"@type": "schema:organization",
"schema:name": "Some org",
"schema:url": "https://www.test.com"
}
}
kr = KR()
kr.set(record)
print(kr)
