def test_save_segment(self): # Update a segment segment = Segment.find(id=self.segment.id) now = datetime.utcnow() updated_name = 'Updated %s' % (time.mktime(now.timetuple())) segment.name = updated_name segment.save() segment = Segment.find(id=self.segment.id) self.assertEqual(segment.name, updated_name)
def test_iterate(self): # Iterate over all segments for segment in Segment.all(): self.assertTrue(segment.id is not None)
def test_find_segment(self): # Find a segment segment = Segment.find(id=self.segment.id) self.assertEqual(segment.id, self.segment.id)
def setup_class(cls): cls.segment = Segment.all()[0]
def test_total_segment_count(self): ok_(Segment.count() >= 0)