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)
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)