def test_get_site_id_from_site(self): site = Site() site.id = 10 self.assertEqual(10, get_site_id(site))