def test_merge04(self):
"""
Test merge with Allen.X_STARTS_Y intervals.
:return:
"""
rows = [{
'a': 1,
'b': 2,
'c': 3,
'start1': '2000-01-01',
'end1': '2000-01-31',
'start2': '2000-01-01',
'end2': '2000-02-29',
'value1': 10.0
}]
expected = [{
'a': 1,
'b': 2,
'c': 3,
'start1': '2000-01-01',
'end1': '2000-01-31',
'value1': 10.0
}]
helper = Type2JoinHelper('start1', 'end1', ['a', 'b', 'c'])
helper.prepare_data(rows)
helper.merge([('start2', 'end2')])
actual = helper.get_rows()
self.assertListEqual(expected, actual)
def test_merge00(self):
"""
Test merge with nor rows.
:return:
"""
rows = []
expected = []
helper = Type2JoinHelper('start1', 'end1', ['a', 'b', 'c'])
helper.prepare_data(rows)
helper.merge([('start2', 'end2')])
actual = helper.get_rows()
self.assertListEqual(expected, actual)
def test_merge27b(self):
"""
Test merge with Allen.X_FINISHES_Y_INVERSE intervals.
:return:
"""
rows = [{
'a': 1,
'b': 2,
'c': 3,
'start': '2000-01-01',
'end': '2000-12-31',
'value1': 10.0
}, {
'a': 1,
'b': 2,
'c': 3,
'start': '2000-12-01',
'end': '2000-12-31',
'value1': 11.0
}]
expected = [{
'a': 1,
'b': 2,
'c': 3,
'start': '2000-01-01',
'end': '2000-11-30',
'value1': 10.0
}, {
'a': 1,
'b': 2,
'c': 3,
'start': '2000-12-01',
'end': '2000-12-31',
'value1': 11.0
}]
helper = Type2JoinHelper('start', 'end', ['a', 'b', 'c'])
helper.prepare_data(rows)
helper.merge([])
actual = helper.get_rows()
self.assertListEqual(expected, actual)
def test_merge23b(self):
"""
Test merge with Allen.X_OVERLAPS_WITH_Y intervals.
:return:
"""
rows = [{
'a': 1,
'b': 2,
'c': 3,
'start': '2000-01-01',
'end': '2000-01-31',
'value1': 10.0
}, {
'a': 1,
'b': 2,
'c': 3,
'start': '2000-01-15',
'end': '2000-02-29',
'value1': 11.0
}]
expected = [{
'a': 1,
'b': 2,
'c': 3,
'start': '2000-01-01',
'end': '2000-01-14',
'value1': 10.0
}, {
'a': 1,
'b': 2,
'c': 3,
'start': '2000-01-15',
'end': '2000-02-29',
'value1': 11.0
}]
helper = Type2JoinHelper('start', 'end', ['a', 'b', 'c'])
helper.prepare_data(rows)
helper.merge([])
actual = helper.get_rows()
self.assertListEqual(expected, actual)
def test_merge14(self):
"""
Test merge with an invalid interval.
:return:
"""
rows = [{
'a': 1,
'b': '2',
'c': 3,
'start1': '2000-01-01',
'end1': '2000-01-31',
'start2': '2000-04-01',
'end2': '2000-03-31',
'value1': 10.0
}]
helper = Type2JoinHelper('start1', 'end1', ['a', 'b', 'c'])
helper.prepare_data(rows)
helper.merge([('start2', 'end2')])
actual = helper.get_rows()
self.assertListEqual([], actual)
def test_merge10(self):
"""
Test merge with Allen.X_OVERLAPS_WITH_Y_INVERSE intervals.
:return:
"""
rows = [{
'a': 1,
'b': 2,
'c': 3,
'start1': '2000-02-01',
'end1': '2000-02-29',
'start2': '2000-01-01',
'end2': '2000-01-31',
'value1': 10.0
}]
helper = Type2JoinHelper('start1', 'end1', ['a', 'b', 'c'])
helper.prepare_data(rows)
helper.merge([('start2', 'end2')])
actual = helper.get_rows()
self.assertListEqual([], actual)
def test_merge99(self):
rows1 = [{
'a': 1,
'b': 1,
'start1': '2000-01-01',
'end1': '2000-01-14',
'value1': 10.0
}, {
'a': 1,
'b': 1,
'start1': '2000-01-15',
'end1': '2000-01-31',
'value1': 10.1
}, {
'a': 1,
'b': 1,
'start1': '2000-02-01',
'end1': '2000-02-29',
'value1': 10.2
}, {
'a': 2,
'b': 3,
'start1': '2000-01-01',
'end1': '2000-01-31',
'value1': 20.0
}, {
'a': 3,
'b': 3,
'start1': '2000-02-01',
'end1': '2000-02-29',
'value1': 20.1
}]
rows2 = [{
'b': 1,
'c': 4,
'start2': '2000-01-01',
'end2': '2000-01-20',
'value2': 1.0
}, {
'b': 1,
'c': 5,
'start2': '2000-01-10',
'end2': '2000-01-14',
'value2': 1.1
}, {
'b': 1,
'c': 6,
'start2': '2000-02-15',
'end2': '2000-02-29',
'value2': 1.2
}, {
'b': 3,
'c': 7,
'start2': '2000-01-01',
'end2': '2000-02-10',
'value2': 2.0
}, {
'b': 3,
'c': 8,
'start2': '2000-02-11',
'end2': '2000-02-20',
'value2': 2.1
}, {
'b': 3,
'c': 9,
'start2': '2000-02-21',
'end2': '2000-03-30',
'value2': 2.2
}]
rows = list()
for row1 in rows1:
for row2 in rows2:
if row1['b'] == row2['b']:
tmp = copy.copy(row1)
tmp.update(row2)
rows.append(tmp)
expected = [{
'a': 1,
'b': 1,
'c': 4,
'end1': '2000-01-14',
'start1': '2000-01-01',
'value1': 10.0,
'value2': 1.0
}, {
'a': 1,
'b': 1,
'c': 4,
'end1': '2000-01-20',
'start1': '2000-01-15',
'value1': 10.1,
'value2': 1.0
}, {
'a': 1,
'b': 1,
'c': 5,
'end1': '2000-01-14',
'start1': '2000-01-10',
'value1': 10.0,
'value2': 1.1
}, {
'a': 1,
'b': 1,
'c': 6,
'end1': '2000-02-29',
'start1': '2000-02-15',
'value1': 10.2,
'value2': 1.2
}, {
'a': 2,
'b': 3,
'c': 7,
'end1': '2000-01-31',
'start1': '2000-01-01',
'value1': 20.0,
'value2': 2.0
}, {
'a': 3,
'b': 3,
'c': 7,
'end1': '2000-02-10',
'start1': '2000-02-01',
'value1': 20.1,
'value2': 2.0
}, {
'a': 3,
'b': 3,
'c': 8,
'end1': '2000-02-20',
'start1': '2000-02-11',
'value1': 20.1,
'value2': 2.1
}, {
'a': 3,
'b': 3,
'c': 9,
'end1': '2000-02-29',
'start1': '2000-02-21',
'value1': 20.1,
'value2': 2.2
}]
helper = Type2JoinHelper('start1', 'end1', ['a', 'b', 'c'])
helper.prepare_data(rows)
helper.merge([('start2', 'end2')])
actual = helper.get_rows()
actual = sorted(actual,
key=lambda row:
(row['a'], row['b'], row['c'], row['start1']))
self.assertListEqual(expected, actual)