def test_copy_from(self):
keyvalcomposer1 = MockKeyValueComposer()
summarizer1 = MockSummarizer()
obj1 = Summary.Reader(keyvalcomposer1, summarizer1)
keyvalcomposer2 = MockKeyValueComposer()
summarizer2 = MockSummarizer()
obj2 = Summary.Reader(keyvalcomposer2, summarizer2)
obj1.copy_from(obj2)
self.assertIs(summarizer2, summarizer1.copy_from_called_with[0])
def test_event(self):
keyvalcomposer = MockKeyValueComposer()
summarizer = MockSummarizer()
weightCalculator = MockWeightCalculator()
obj = Summary.Reader(keyvalcomposer,
summarizer,
weightCalculator=weightCalculator)
# two key-val pairs
key1 = MockKey('key1')
key2 = MockKey('key2')
val1 = MockVal('val1')
val2 = MockVal('val2')
event = MockEvent(event='event1', keys=(key1, key2), vals=(val1, val2))
obj.event(event)
self.assertEqual([
(key1, val1, MockWeight(event)),
(key2, val2, MockWeight(event)),
], summarizer.add_called_with)
# no key-val pairs
summarizer.add_called_with[:] = []
event = MockEvent(event='event1', keys=(), vals=())
obj.event(event)
self.assertEqual([], summarizer.add_called_with)
def test_event_nevents(self):
keyvalcomposer = MockKeyValueComposer()
summarizer = MockSummarizer()
weightCalculator = MockWeightCalculator()
obj = Summary.Reader(
keyvalcomposer,
summarizer,
weightCalculator=weightCalculator,
nevents=2 # read only first 2 events
)
key1 = MockKey('key1')
val1 = MockVal('val1')
event1 = MockEvent(event='event1', keys=(key1, ), vals=(val1, ))
event2 = MockEvent(event='event2', keys=(key1, ), vals=(val1, ))
event3 = MockEvent(event='event3', keys=(key1, ), vals=(val1, ))
obj.event(event1)
obj.event(event2)
obj.event(event3)
self.assertEqual([
(key1, val1, MockWeight(event1)),
(key1, val1, MockWeight(event2)),
], summarizer.add_called_with)
def test_begin(self):
keyvalcomposer = MockKeyValueComposer()
summarizer = MockSummarizer()
obj = Summary.Reader(keyvalcomposer, summarizer)
self.assertIsNone(keyvalcomposer.began_with)
event = MockEvent(event='event1', keys=(), vals=())
obj.begin(event)
self.assertEqual(event, keyvalcomposer.began_with)
def test_end_None_nextKeyComposer(self):
key1 = MockKey('key1')
key2 = MockKey('key2')
key3 = MockKey('key3')
keyvalcomposer = MockKeyValueComposer()
summarizer = MockSummarizer()
summarizer.keys_return[:] = [key1, key2, key3]
nextKeyComposer = None
obj = Summary.Reader(keyvalcomposer,
summarizer,
nextKeyComposer=nextKeyComposer)
obj.end()
self.assertEqual(set([]), set(summarizer.add_key_called_with))
def test_end(self):
key1 = MockKey('key1')
key11 = MockKey('key11')
key2 = MockKey('key2')
key21 = MockKey('key21')
key22 = MockKey('key22')
key3 = MockKey('key3')
nextdic = {
key1: (key11, ),
key2: (key21, key22),
key3: (),
}
keyvalcomposer = MockKeyValueComposer()
summarizer = MockSummarizer()
summarizer.keys_return[:] = [key1, key2, key3]
nextKeyComposer = MockNextKeyComposer(nextdic)
obj = Summary.Reader(keyvalcomposer,
summarizer,
nextKeyComposer=nextKeyComposer)
obj.end()
self.assertEqual(set([key11, key21, key22]),
set(summarizer.add_key_called_with))
def test_results(self):
keyvalcomposer = MockKeyValueComposer()
summarizer = MockSummarizer()
obj = Summary.Reader(keyvalcomposer, summarizer)
self.assertIs(summarizer, obj.results())