def test_FindorAddCompany(self):
n = Network()
companyid = "company"
companyid3 = "company3"
c = Company(companyid)
n.AddCompany(c)
self.assertTrue(n.FindOrAddCompany(companyid).ID, companyid)
self.assertTrue(n.FindOrAddCompany(companyid3).ID, companyid3)
self.assertTrue(n.Companies[companyid3].ID, companyid3)
return
