def test_6_event_query(self): # get_event without limit and type pt_1 = GeniusEventQuery.get_event(self.session) assert_result = len(list(pt_1)) self.assertEqual(assert_result, 20) # get_event with limit pt_2 = GeniusEventQuery.get_event(self.session, None, 10) assert_result = len(list(pt_2)) self.assertEqual(assert_result, 10) # get_event with type pt_3 = GeniusEventQuery.get_event(self.session, "add") for pt in pt_3: self.assertEqual(pt.event_type, "add") pt.event_type = "more" self.assertEqual(pt.event_type, "more")