def setUp(self): self.albums = Marble(name="Albums", fields=_FIELDS, partition=_PARTITIONS) self.transaction = Marble(name="Transcation", fields=_FIELDS_SELL, partition=_PARTITIONS) self.single_where = [(self.albums.rating > 3)] self.multi_wheres = [ (self.albums.rating > 3) & (self.albums.id == 1000) ] self.cross_wheres = [ self.albums.rating > 3, self.transaction.id == 1000 ] self.single_select = [self.albums.name] self.multi_select = [ self.albums.name, self.albums.date, self.albums.rating ] self.cross_select = [ self.albums.name, self.albums.artist, self.transaction.store_id, self.transaction.price ] self.order_by = [self.albums.quantity, self.albums.rating] self.join = [self.albums.id, self.transaction.item_id] self.join_invalid = [self.albums.id, self.transaction.price] self.join_invalid_1 = [self.albums.id, self.albums.id] self.join_invalid_2 = [self.albums.id, self.transaction.price] self.limit_single = 100 self.limit_single_invalid = -100
def setUp(self): self.albums = [dict(zip(_FIELDS_RAW, album)) for album in _ALBUMS] self.marble = Marble(name="Collections", fields=_FIELDS, partition=_PARTITIONS) self.n_inserted, self.files = self.marble._insert([ (ujson.dumps(l) for l in self.albums) ])
def test_partition_numbers(self): self.albums = [dict(zip(_FIELDS_RAW, album)) for album in _ALBUMS] self.marble = Marble(name="Collections", fields=_FIELDS, partition=_PARTITIONS) self.n_inserted, self.files = self.marble._insert( [(ujson.dumps(l) for l in self.albums)], partition_filter='1992-10-03') self.assertEquals(len(self.files), 1) inserted = len( [1 for album in self.albums if album['date'] == '1992-10-03']) self.assertEquals(inserted, self.n_inserted) for date, file in self.files.iteritems(): os.unlink(file)
def setUp(self): self.emp = Marble(name="employee", fields=EMP_FIELDS) self.dept = Marble(name="department", fields=DEPT_FIELDS)