def testReadData(self):
db = edb.get_fake_trips_db()
trips = db.find()
uuids = set()
for t in trips:
uuids.add(t['user_id'])
if len(uuids) == 0:
uuids.add(None)
self.uuids = uuids
data = cp.read_data(size=10)
print 'there are ' + str(len(data))
self.assertTrue(len(data) == 10 or len(data) == 0) #len(data)==0 if the test is run on an empty database
uuid = 'baduuid'
data = cp.read_data(uuid=uuid)
self.assertTrue(not data)
uuid = self.uuids.pop()
data = cp.read_data(uuid = uuid)
self.assertTrue(len(data) <= db.find({'user_id' : uuid}).count())
self.uuids.add(uuid)
sum = 0
for uuid in self.uuids:
data = cp.read_data(uuid=uuid)
sum += len(data)
data = cp.read_data()
self.assertTrue(0 <= len(data) <= db.find().count())
self.assertTrue(len(data) == sum)
for d in data:
self.assertTrue(d.trip_start_location and d.trip_end_location and d.start_time)
def save_to_db(self):
db = edb.get_fake_trips_db()
db.insert({
"_id":
self._id,
"user_id":
self.user_id,
"trip_id":
self.trip_id,
"sections":
range(len(self.sections)),
"trip_start_datetime":
self.start_time,
"trip_end_datetime":
self.end_time,
"trip_start_location":
self.trip_start_location.coordinate_list(),
"trip_end_location":
self.trip_end_location.coordinate_list(),
"mode_list":
self.mode_list
})
def save_to_db(self):
db = edb.get_fake_trips_db()
print "trip start loc is %s" % self.trip_start_location
print "trip end loc is %s" % self.trip_end_location
db.insert({"trip_id" : self._id, "trip_start_location" : self.trip_start_location.coordinate_list(),
"trip_end_location" : self.trip_end_location.coordinate_list()})
def save_to_db(self):
db = edb.get_fake_trips_db()
db.insert({"_id": self._id, "user_id": self.user_id, "trip_id": self.trip_id, "sections": list(range(len(self.sections))), "trip_start_datetime": self.start_time,
"trip_end_datetime": self.end_time, "trip_start_location": self.trip_start_location.coordinate_list(),
"trip_end_location": self.trip_end_location.coordinate_list(), "mode_list": self.mode_list})
