def test_FindOrAddStation(self): n = Network() stationid = "station" stationid3 = "station3" companyid = "company" divisionid = "division" s = Station(stationid) s2 = Station(stationid) n.AddStation(s) self.assertEqual(n.FindOrAddStation(stationid).ID, stationid) self.assertEqual(n.FindOrAddStation(stationid3,companyid, divisionid).ID, stationid3) self.assertEqual(n.Stations[stationid3].ID, stationid3) self.assertEqual(n.Companies[companyid].Stations[stationid3].ID, stationid3) self.assertEqual(n.Companies[companyid].Divisions[divisionid].Stations[stationid3].ID, stationid3) self.assertEqual(n.Stations[stationid3].Company.ID, companyid) self.assertEqual(n.Stations[stationid3].Division.ID, divisionid) return