예제 #1
0
    def testInsert(self):
        table = Table('census', self.schema).data(self.df)
        c1 = table.row_count()
        table.insert({'State': 'ES', 'life_meaning':42})
        self.assertEqual(table.row_count() - c1, 1)

        c2 = table.row_count()
        table.insert([{'State': 'ES', 'life_meaning':42},
                      {'State': 'ES2', 'life_meaning':42},])
        self.assertEqual(table.row_count() - c2, 2)

        view = table.find({'life_meaning': {'$exists':True}})
        self.assertEqual(view.row_count(), 3)
        self.assertEqual(table.find_one({'life_meaning': {'$exists':True}})['life_meaning'], 42)
예제 #2
0
 def testRowCount(self):
     table = Table('census', self.schema).data(self.df)
     self.assertEqual(table.row_count(), 51)
     view = table.find({'$or':[{'State': 'NY'},{'State': 'DC'}]})
     self.assertEqual(view.row_count(), 2)
     view2 = view.find({'State': 'NY'})
     self.assertEqual(view2.row_count(), 1)