def test_join_things(self): fields = ["url", "sr_id"] link = make_link(thing_id=1, timestamp=100.01, url="foo", sr_id=10) dump = make_pg_dump(fields, link) stdin = StringIO(dump) stdout = StringIO() mr_tools.join_things(fields, fd=stdin, out=stdout, err=StringIO()) self.assertEqual(stdout.getvalue().strip(), "1 link 1 0 f f 100.01 foo 10".replace(" ", "\t"))
def join_things(self): fields = [] defaults = {} for k, typ in self.fields.iteritems(): fields.append(k) if typ is int: defaults[k] = 0 elif typ is str: defaults[k] = "" mr_tools.join_things( fields, fd=self.fd, out=self.out, err=self.err, defaults=defaults )
def join_comments(): return mr_tools.join_things(('link_id',))
def join_links(): mr_tools.join_things(('url',))
def join_things(thing_type): mr_tools.join_things(data_fields_by_name[thing_type].keys())
def join_comments(): join_things(('author_id',))
def join_links(): join_things(("author_id", "sr_id"))
def join_links(): mr_tools.join_things(('author_id',))
def join_links(): mr_tools.join_things(('url', ))
def join_things(): mr_tools.join_things(('url', 'sr_id', 'author_id'))
def join_links(): mr_tools.join_things(('author_id', ))
def join_comments(): return mr_tools.join_things(('link_id', ))
def join_comments(): join_things(('author_id', ))
def join_links(): join_things(('author_id', 'sr_id'))
def join_links(): mr_tools.join_things(('url', 'sr_id'))
def join_comments(): join_things(("author_id",))