diff --git a/tests/api/test_api.py b/tests/api/test_api.py index b58e2b7ad..c41191788 100644 --- a/tests/api/test_api.py +++ b/tests/api/test_api.py @@ -59,7 +59,7 @@ def make_records(**kwargs): for record_type in kwargs.get('record_type', []): if record_type in GENERATORS.keys(): for values in algebraic_product(**kwargs): - record = GENERATORS[record_type]().next() + record = next(GENERATORS[record_type]()) record.update(values) yield record diff --git a/tests/unit/test_record_processor.py b/tests/unit/test_record_processor.py index b09588172..39902c8f2 100644 --- a/tests/unit/test_record_processor.py +++ b/tests/unit/test_record_processor.py @@ -545,8 +545,8 @@ class TestRecordProcessor(testtools.TestCase): 'company_draft': 'Mirantis'} record_processor_inst = self.make_record_processor() - result_member = record_processor_inst._process_member( - member_record).next() + result_member = next(record_processor_inst._process_member( + member_record)) self.assertEqual(result_member['primary_key'], 'member:123456789') self.assertEqual(result_member['date'], utils.member_date_to_timestamp( @@ -576,8 +576,8 @@ class TestRecordProcessor(testtools.TestCase): updated_member_record['member_name'] = 'Bill Smith' updated_member_record['company_draft'] = 'Rackspace' - result_member = record_processor_inst._process_member( - updated_member_record).next() + result_member = next(record_processor_inst._process_member( + updated_member_record)) self.assertEqual(result_member['author_name'], 'Bill Smith') self.assertEqual(result_member['company_name'], 'Rackspace')