def test():
from r2.lib.mr_tools._mr_tools import keyiter
for key, vals in keyiter():
print key, vals
for val in vals:
print "\t", val
def test():
from r2.lib.mr_tools._mr_tools import keyiter
for key, vals in keyiter():
print key, vals
for val in vals:
print '\t', val
def test_keyiter(self):
stdin = StringIO("\n".join([
"foo\tbar\tbar1",
"baz\tbad\tbad1",
]))
d = {k: list(v) for k, v in keyiter(stream=stdin)}
self.assertEqual(d, {
"foo": [["bar", "bar1"]],
"baz": [["bad", "bad1"]],
})
def test_keyiter(self):
stdin = StringIO("\n".join(["foo\tbar\tbar1", "baz\tbad\tbad1"]))
d = {k: list(v) for k, v in keyiter(stream=stdin)}
self.assertEqual(d, {"foo": [["bar", "bar1"]], "baz": [["bad", "bad1"]]})