def test_subjects_xml(self, item, expected):
filename = os.path.dirname(__file__) + '/test_data/xml_input/' + item + '_marc.xml'
element = etree.parse(filename).getroot()
if element.tag != record_tag and element[0].tag == record_tag:
element = element[0]
rec = MarcXml(element)
assert read_subjects(rec) == expected
def test_subjects_bin(self, item, expected):
filename = os.path.dirname(__file__) + '/test_data/bin_input/' + item
data = open(filename).read()
if len(data) != int(data[:5]):
data = data.decode('utf-8').encode('raw_unicode_escape')
rec = MarcBinary(data)
assert read_subjects(rec) == expected
def _test_subjects(self, rec, expect):
print read_subjects(rec)
assert read_subjects(rec) == expect
data = open(filename).read()
if len(data) != int(data[:5]):
data = data.decode('utf-8').encode('raw_unicode_escape')
rec = MarcBinary(data)
yield self._test_subjects, rec, expect
subjects = []
for item, expect in xml_samples:
filename = 'test_data/' + item + '_marc.xml'
element = etree.parse(filename).getroot()
if element.tag != record_tag and element[0].tag == record_tag:
element = element[0]
rec = MarcXml(element)
subjects.append(read_subjects(rec))
for item, expect in bin_samples:
filename = 'test_data/' + item
data = open(filename).read()
if len(data) != int(data[:5]):
data = data.decode('utf-8').encode('raw_unicode_escape')
rec = MarcBinary(data)
subjects.append(read_subjects(rec))
all_subjects = defaultdict(lambda: defaultdict(int))
for a in subjects:
for b, c in a.items():
for d, e in c.items():
all_subjects[b][d] += e
def _test_subjects(self, rec, expect):
print read_subjects(rec)
assert read_subjects(rec) == expect
data = open(filename).read()
if len(data) != int(data[:5]):
data = data.decode("utf-8").encode("raw_unicode_escape")
rec = MarcBinary(data)
yield self._test_subjects, rec, expect
subjects = []
for item, expect in xml_samples:
filename = "test_data/" + item + "_marc.xml"
element = etree.parse(filename).getroot()
if element.tag != record_tag and element[0].tag == record_tag:
element = element[0]
rec = MarcXml(element)
subjects.append(read_subjects(rec))
for item, expect in bin_samples:
filename = "test_data/" + item
data = open(filename).read()
if len(data) != int(data[:5]):
data = data.decode("utf-8").encode("raw_unicode_escape")
rec = MarcBinary(data)
subjects.append(read_subjects(rec))
all_subjects = defaultdict(lambda: defaultdict(int))
for a in subjects:
for b, c in a.items():
for d, e in c.items():
all_subjects[b][d] += e
