def test_pull_requests(self): pr1 = pullrequest_factory("PR1") pr2 = pullrequest_factory("PR2") pr3 = pullrequest_factory("PR3") req1 = Requirement.parse("django", 0) req1.pull_request = pr1 req2 = Requirement.parse("pyramid", 0) req2.pull_request = pr1 req3 = Requirement.parse("flask", 0) req3.pull_request = pr2 req4 = Requirement.parse("elixir", 0) req4.pull_request = pr3 reqs = RequirementsBundle() req_file = RequirementFile(path="foo", content='') req_file._requirements = [ req1, req2, req3, req4 ] reqs.append(req_file) self.assertEqual( [pr1, pr2, pr3], [pr for pr in reqs.pull_requests()] )