コード例 #1
0
ファイル: test_query_checker.py プロジェクト: zhengyuv/hustle
 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
コード例 #2
0
 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)
     ])
コード例 #3
0
ファイル: test_marble.py プロジェクト: zhengyuv/hustle
    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)
コード例 #4
0
ファイル: test_pipeline.py プロジェクト: zhengyuv/hustle
 def setUp(self):
     self.emp = Marble(name="employee", fields=EMP_FIELDS)
     self.dept = Marble(name="department", fields=DEPT_FIELDS)