def test_FindorAddDivision(self): n = Network() companyid = "company" companyid3 = "company3" divisionid = "division" divisionid3 = "division3" c = Company(companyid) d = Division(divisionid, companyid) n.AddCompany(c) n.AddDivision(d) self.assertTrue(n.FindOrAddDivision(divisionid, companyid).ID, divisionid) self.assertTrue(n.FindOrAddDivision(divisionid3, companyid3).ID, divisionid3) self.assertTrue(n.Companies[companyid3].Divisions[divisionid3].ID, divisionid3) return