def testing_truncate_sites_for_test(self): attributes = {"mo": array(["money", "soup"]), "SITE_CLASS": array(["E", "C"]), "id": array([1, 2])} latitude = [10, 20] longitude = [1, 2] sites = Sites(latitude, longitude, **attributes) use_site_indexes = False site_indexes = array([2]) new_sites = truncate_sites_for_test(use_site_indexes, sites, site_indexes) self.failUnless(allclose(array([1, 2]), new_sites.attributes["id"])) use_site_indexes = True site_indexes = array([2]) new_sites = truncate_sites_for_test(use_site_indexes, sites, site_indexes) self.failUnlessEqual(site_indexes, new_sites.attributes["id"])
def testing_truncate_sites_for_test(self): attributes = {'mo': array(['money', 'soup']), 'SITE_CLASS': array(['E', 'C']), 'id': array([1, 2])} latitude = [10, 20] longitude = [1, 2] sites = Sites(latitude, longitude, **attributes) use_site_indexes = False site_indexes = array([2]) new_sites = truncate_sites_for_test(use_site_indexes, sites, site_indexes) self.failUnless(allclose(array([1, 2]), new_sites.attributes['id'])) use_site_indexes = True site_indexes = array([2]) new_sites = truncate_sites_for_test(use_site_indexes,sites, site_indexes) self.failUnlessEqual(site_indexes, new_sites.attributes['id'])