hangshuo652
19 lines
636 B
Python
19 lines
636 B
Python
import sys, os
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
|
|
from comparator.aligner import align_records
|
|
|
|
|
|
def test_align_large_key_set():
|
|
cobol = [{"ID": f"K{i:04d}", "V": i} for i in range(100)]
|
|
spark = [{"ID": f"K{i:04d}", "V": i} for i in range(100)]
|
|
result = align_records(cobol, spark, key_field="ID")
|
|
assert len(result) == 100
|
|
assert all(st == "MATCHED" for _, _, st in result)
|
|
|
|
|
|
def test_align_empty_key_value():
|
|
cobol = [{"ID": "", "V": 1}]
|
|
spark = [{"ID": "", "V": 1}]
|
|
result = align_records(cobol, spark, key_field="ID")
|
|
assert len(result) == 1
|