From 4598d5ed7df56d164c59aa0a9e332e7f5be8378f Mon Sep 17 00:00:00 2001 From: antoniocarlon Date: Fri, 14 Sep 2018 10:14:59 +0200 Subject: [PATCH] Added CR suggestion --- test/integration/test_admin0_functions.py | 4 +- test/integration/test_admin1_functions.py | 4 +- .../test_data_observatory_functions.py | 52 +++++++++---------- test/integration/test_ipaddress_functions.py | 4 +- test/integration/test_isolines_functions.py | 6 +-- test/integration/test_namedplace_functions.py | 4 +- test/integration/test_postalcode_functions.py | 4 +- test/integration/test_routing_functions.py | 6 +-- 8 files changed, 42 insertions(+), 42 deletions(-) diff --git a/test/integration/test_admin0_functions.py b/test/integration/test_admin0_functions.py index 9d3dd79..7e98abe 100644 --- a/test/integration/test_admin0_functions.py +++ b/test/integration/test_admin0_functions.py @@ -1,6 +1,6 @@ from unittest import TestCase from nose.tools import assert_raises -from nose.tools import assert_not_equal, assert_true +from nose.tools import assert_not_equal, assert_in from ..helpers.integration_test_helper import IntegrationTestHelper @@ -28,4 +28,4 @@ class TestAdmin0Functions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Geocoding permission denied", "function cdb_geocode_admin0_polygon(text) does not exist"]) + assert_in(e.message[0], ["Geocoding permission denied", "function cdb_geocode_admin0_polygon(text) does not exist"]) diff --git a/test/integration/test_admin1_functions.py b/test/integration/test_admin1_functions.py index edfe19e..92ffb41 100644 --- a/test/integration/test_admin1_functions.py +++ b/test/integration/test_admin1_functions.py @@ -1,6 +1,6 @@ from unittest import TestCase from nose.tools import assert_raises -from nose.tools import assert_not_equal, assert_true +from nose.tools import assert_not_equal, assert_in from ..helpers.integration_test_helper import IntegrationTestHelper @@ -37,4 +37,4 @@ class TestAdmin1Functions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Geocoding permission denied", "function cdb_geocode_admin1_polygon(text) does not exist"]) + assert_in(e.message[0], ["Geocoding permission denied", "function cdb_geocode_admin1_polygon(text) does not exist"]) diff --git a/test/integration/test_data_observatory_functions.py b/test/integration/test_data_observatory_functions.py index 00e916b..661678a 100644 --- a/test/integration/test_data_observatory_functions.py +++ b/test/integration/test_data_observatory_functions.py @@ -1,6 +1,6 @@ from unittest import TestCase from nose.tools import assert_raises -from nose.tools import assert_not_equal, assert_true +from nose.tools import assert_not_equal, assert_in from ..helpers.integration_test_helper import IntegrationTestHelper @@ -24,7 +24,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getdemographicsnapshot(geometry) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getdemographicsnapshot(geometry) does not exist"]) def test_if_get_segment_snapshot_is_ok(self): query = "SELECT OBS_GetSegmentSnapshot(CDB_LatLng(40.704512, -73.936669)) as snapshot;&api_key={0}".format(self.env_variables['api_key']) @@ -36,7 +36,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getsegmentsnapshot(geometry) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getsegmentsnapshot(geometry) does not exist"]) def test_if_get_measure_with_point_is_ok(self): query = "SELECT OBS_GetMeasure(CDB_LatLng(40.704512, -73.936669), 'us.census.acs.B01003001') as measure;&api_key={0}".format(self.env_variables['api_key']) @@ -53,7 +53,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getmeasure(geometry, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getmeasure(geometry, unknown) does not exist"]) def test_if_get_measure_by_id_ok(self): query = "SELECT OBS_GetMeasureById('36047048500', 'us.census.acs.B01003001', 'us.census.tiger.census_tract', '2010 - 2014') as measure;&api_key={0}".format(self.env_variables['api_key']) @@ -65,7 +65,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getmeasurebyid(unknown, unknown, unknown, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getmeasurebyid(unknown, unknown, unknown, unknown) does not exist"]) def test_if_get_category_is_ok(self): query = "SELECT OBS_GetCategory(CDB_LatLng(40.704512, -73.936669), 'us.census.spielman_singleton_segments.X10', 'us.census.tiger.census_tract', '2010 - 2014') as category;&api_key={0}".format(self.env_variables['api_key']) @@ -77,7 +77,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getcategory(geometry, unknown, unknown, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getcategory(geometry, unknown, unknown, unknown) does not exist"]) def test_if_get_us_census_measure_with_point_is_ok(self): query = "SELECT OBS_GetUSCensusMeasure(CDB_LatLng(40.704512, -73.936669), 'male population') as measure;&api_key={0}".format(self.env_variables['api_key']) @@ -94,7 +94,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getuscensusmeasure(geometry, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getuscensusmeasure(geometry, unknown) does not exist"]) def test_if_get_us_census_category_is_ok(self): query = "SELECT OBS_GetUSCensusCategory('0103000020E61000000100000021000000BB366F6D917B52C0E7BB6EA82B5A444067224C85937B52C0810205E70E5A4440596D6342997B52C09BED2952F35944400A96386CA27B52C07B6D31F9D95944402B81A0A8AE7B52C0D16B73D5C3594440C34C397FBD7B52C0D8B5B7C0B1594440CFD90A5ECE7B52C0BE8DD86CA45944405C7E229FE07B52C0A904EE5C9C59444017C1F28EF37B52C05E2745E099594440AE8A3873067C52C0FD62540F9D5944402F272292187C52C05502CBCAA5594440C3F17139297C52C0B3FBC4BCB3594440DDAE56C5377C52C00C46175CC6594440E96AB6A6437C52C0355C94F1DC5944402313AE684C7C52C05340159FF6594440B3B90FB5517C52C031F30168125A44407F43B357537C52C0DF95053B2F5A444059C17840517C52C0D9E08EFC4B5A44406580E8834B7C52C046B5B591675A444035736A5A427C52C0AF9A1AEB805A44402E721C1E367C52C0D226550F975A4440506D5C47277C52C0A2968A24A95A4440507C2868167C52C0D51FCE78B65A4440C4438226047C52C02369F888BE5A4440C9F10C36F17B52C027B1B205C15A4440AABE2451DE7B52C0F5E383D6BD5A44402A18B431CC7B52C01785C11AB55A444018320D8ABB7B52C070265B28A75A4440E28A0EFEAC7B52C09E518C88945A44401D08D61CA17B52C09E8195F27D5A4440D7C3405B987B52C0643CB044645A444047B16D0F937B52C04EC9837B485A4440BB366F6D917B52C0E7BB6EA82B5A4440'::geometry, 'Spielman-Singleton Segments: 10 Clusters') as category;&api_key={0}".format(self.env_variables['api_key']) @@ -106,7 +106,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getuscensuscategory(geometry, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getuscensuscategory(geometry, unknown) does not exist"]) def test_if_get_population_is_ok(self): query = "SELECT OBS_GetPopulation(CDB_LatLng(40.704512, -73.936669)) as population;&api_key={0}".format(self.env_variables['api_key']) @@ -118,7 +118,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getpopulation(geometry) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getpopulation(geometry) does not exist"]) def test_if_obs_search_is_ok(self): sql = "SELECT id FROM OBS_Search('total_pop') WHERE id LIKE 'es.ine%' LIMIT 1;" @@ -132,7 +132,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_search(unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_search(unknown) does not exist"]) def test_if_obs_get_available_boundaries_is_ok(self): query = "SELECT boundary_id FROM OBS_GetAvailableBoundaries(CDB_LatLng(40.704512, -73.936669)) LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -144,7 +144,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getavailableboundaries(geometry) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getavailableboundaries(geometry) does not exist"]) def test_if_obs_get_boundary_is_ok(self): query = "SELECT OBS_GetBoundary(CDB_LatLng(40.704512, -73.936669), 'us.census.tiger.census_tract') as boundary;&api_key={0}".format(self.env_variables['api_key']) @@ -156,7 +156,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getboundary(geometry, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getboundary(geometry, unknown) does not exist"]) def test_if_obs_get_boundary_id_is_ok(self): query = "SELECT OBS_GetBoundaryId(CDB_LatLng(40.704512, -73.936669), 'us.census.tiger.census_tract', '2015') as boundary_id;&api_key={0}".format(self.env_variables['api_key']) @@ -168,7 +168,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getboundaryid(geometry, unknown, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getboundaryid(geometry, unknown, unknown) does not exist"]) def test_if_obs_get_boundary_by_id_is_ok(self): query = "SELECT OBS_GetBoundaryById('36047', 'us.census.tiger.county', '2014') as boundary;&api_key={0}".format(self.env_variables['api_key']) @@ -180,7 +180,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getboundarybyid(unknown, unknown, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getboundarybyid(unknown, unknown, unknown) does not exist"]) def test_if_obs_get_boundaries_by_geometry_is_ok(self): query = "SELECT geom_refs FROM OBS_GetBoundariesByGeometry(ST_MakeEnvelope(-73.9452409744,40.6988851644,-73.9280319214,40.7101254524,4326), 'us.census.tiger.census_tract') ORDER BY geom_refs ASC LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -192,7 +192,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getboundariesbygeometry(geometry, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getboundariesbygeometry(geometry, unknown) does not exist"]) def test_if_obs_get_boundaries_by_point_and_radius_is_ok(self): query = "SELECT geom_refs FROM OBS_GetBoundariesByPointAndRadius(CDB_LatLng(40.704512, -73.936669), 500, 'us.census.tiger.census_tract') ORDER BY geom_refs ASC LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -204,7 +204,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getboundariesbypointandradius(geometry, integer, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getboundariesbypointandradius(geometry, integer, unknown) does not exist"]) def test_if_obs_get_points_by_geometry_is_ok(self): query = "SELECT geom_refs FROM OBS_GetPointsByGeometry(ST_MakeEnvelope(-73.9452409744,40.6988851644,-73.9280319214,40.7101254524,4326), 'us.census.tiger.census_tract') ORDER BY geom_refs ASC LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -216,7 +216,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getpointsbygeometry(geometry, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getpointsbygeometry(geometry, unknown) does not exist"]) def test_if_obs_get_points_by_point_and_radius_is_ok(self): query = "SELECT geom_refs FROM OBS_GetPointsByPointAndRadius(CDB_LatLng(40.704512, -73.936669), 500, 'us.census.tiger.census_tract', '2014') ORDER BY geom_refs ASC LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -228,7 +228,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getpointsbypointandradius(geometry, integer, unknown, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getpointsbypointandradius(geometry, integer, unknown, unknown) does not exist"]) def test_if_obs_get_legacy_builder_metadata_is_ok(self): query = "SELECT name FROM OBS_LegacyBuilderMetadata() LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -240,7 +240,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_legacybuildermetadata() does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_legacybuildermetadata() does not exist"]) def test_if_obs_get_available_numerators_is_ok(self): query = "SELECT numer_id FROM OBS_GetAvailableNumerators() LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -252,7 +252,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getavailablenumerators() does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getavailablenumerators() does not exist"]) def test_if_obs_get_available_denominators_is_ok(self): query = "SELECT denom_id FROM OBS_GetAvailableDenominators() LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -264,7 +264,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getavailabledenominators() does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getavailabledenominators() does not exist"]) def test_if_obs_get_available_geometries_is_ok(self): query = "SELECT geom_id FROM OBS_GetAvailableGeometries() LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -276,7 +276,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getavailablegeometries() does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getavailablegeometries() does not exist"]) def test_if_obs_get_available_timespans_is_ok(self): query = "SELECT timespan_id FROM OBS_GetAvailableTimespans() LIMIT 1;&api_key={0}".format(self.env_variables['api_key']) @@ -288,7 +288,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getavailabletimespans() does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getavailabletimespans() does not exist"]) def test_if_obs_get_meta_is_ok(self): params = '\'[{\"numer_id\": \"us.census.acs.B01003001\"}]\'' @@ -302,7 +302,7 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getmeta(geometry, unknown, integer, integer, integer) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getmeta(geometry, unknown, integer, integer, integer) does not exist"]) def test_if_obs_get_data_is_ok(self): params_1 = '\'[{\"numer_id\": \"us.census.acs.B01003001\", \"geom_id\": \"us.census.tiger.county\"}]\'' @@ -322,8 +322,8 @@ class TestDataObservatoryFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query_1) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getmeta(geometry, unknown, integer, integer, integer) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getmeta(geometry, unknown, integer, integer, integer) does not exist"]) try: IntegrationTestHelper.execute_query(self.sql_api_url, query_2) except Exception as e: - assert_true(e.message[0] in ["Observatory permission denied", "function obs_getmeta(geometry, unknown) does not exist"]) + assert_in(e.message[0], ["Observatory permission denied", "function obs_getmeta(geometry, unknown) does not exist"]) diff --git a/test/integration/test_ipaddress_functions.py b/test/integration/test_ipaddress_functions.py index 5a06916..191d7d7 100644 --- a/test/integration/test_ipaddress_functions.py +++ b/test/integration/test_ipaddress_functions.py @@ -1,6 +1,6 @@ from unittest import TestCase from nose.tools import assert_raises -from nose.tools import assert_not_equal, assert_true +from nose.tools import assert_not_equal, assert_in from ..helpers.integration_test_helper import IntegrationTestHelper @@ -29,4 +29,4 @@ class TestPostalcodeFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Geocoding permission denied", "function cdb_geocode_ipaddress_point(text) does not exist"]) + assert_in(e.message[0], ["Geocoding permission denied", "function cdb_geocode_ipaddress_point(text) does not exist"]) diff --git a/test/integration/test_isolines_functions.py b/test/integration/test_isolines_functions.py index 6622b40..8da7e00 100644 --- a/test/integration/test_isolines_functions.py +++ b/test/integration/test_isolines_functions.py @@ -1,6 +1,6 @@ from unittest import TestCase from nose.tools import assert_raises -from nose.tools import assert_not_equal, assert_true +from nose.tools import assert_not_equal, assert_in from ..helpers.integration_test_helper import IntegrationTestHelper @@ -27,7 +27,7 @@ class TestIsolinesFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Isolines permission denied", "function cdb_isochrone(geometry, unknown, integer[]) does not exist"]) + assert_in(e.message[0], ["Isolines permission denied", "function cdb_isochrone(geometry, unknown, integer[]) does not exist"]) def test_if_select_with_isodistance_is_ok(self): query = "SELECT * FROM cdb_isodistance('POINT(-3.70568 40.42028)'::geometry, " \ @@ -42,4 +42,4 @@ class TestIsolinesFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Isolines permission denied", "function cdb_isodistance(geometry, unknown, integer[]) does not exist"]) + assert_in(e.message[0], ["Isolines permission denied", "function cdb_isodistance(geometry, unknown, integer[]) does not exist"]) diff --git a/test/integration/test_namedplace_functions.py b/test/integration/test_namedplace_functions.py index 6ed87d4..6d64fd5 100644 --- a/test/integration/test_namedplace_functions.py +++ b/test/integration/test_namedplace_functions.py @@ -1,6 +1,6 @@ from unittest import TestCase from nose.tools import assert_raises -from nose.tools import assert_not_equal, assert_true +from nose.tools import assert_not_equal, assert_in from ..helpers.integration_test_helper import IntegrationTestHelper @@ -45,4 +45,4 @@ class TestNameplaceFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Geocoding permission denied", "function cdb_geocode_namedplace_point(text) does not exist"]) + assert_in(e.message[0], ["Geocoding permission denied", "function cdb_geocode_namedplace_point(text) does not exist"]) diff --git a/test/integration/test_postalcode_functions.py b/test/integration/test_postalcode_functions.py index 909d4a3..13151af 100644 --- a/test/integration/test_postalcode_functions.py +++ b/test/integration/test_postalcode_functions.py @@ -1,6 +1,6 @@ from unittest import TestCase from nose.tools import assert_raises -from nose.tools import assert_not_equal, assert_true +from nose.tools import assert_not_equal, assert_in from ..helpers.integration_test_helper import IntegrationTestHelper @@ -37,4 +37,4 @@ class TestPostalcodeFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Geocoding permission denied", "function cdb_geocode_postalcode_polygon(text, text) does not exist"]) \ No newline at end of file + assert_in(e.message[0], ["Geocoding permission denied", "function cdb_geocode_postalcode_polygon(text, text) does not exist"]) \ No newline at end of file diff --git a/test/integration/test_routing_functions.py b/test/integration/test_routing_functions.py index d6a9ea9..6e9ad65 100644 --- a/test/integration/test_routing_functions.py +++ b/test/integration/test_routing_functions.py @@ -1,6 +1,6 @@ from unittest import TestCase from nose.tools import assert_raises -from nose.tools import assert_not_equal, assert_true +from nose.tools import assert_not_equal, assert_in from ..helpers.integration_test_helper import IntegrationTestHelper @@ -31,7 +31,7 @@ class TestRoutingFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Routing permission denied", "function cdb_route_point_to_point(geometry, geometry, unknown, text[]) does not exist"]) + assert_in(e.message[0], ["Routing permission denied", "function cdb_route_point_to_point(geometry, geometry, unknown, text[]) does not exist"]) def test_if_select_with_routing_with_waypoints_is_ok(self): query = "SELECT duration, length, shape as the_geom " \ @@ -53,4 +53,4 @@ class TestRoutingFunctions(TestCase): try: IntegrationTestHelper.execute_query(self.sql_api_url, query) except Exception as e: - assert_true(e.message[0] in ["Routing permission denied", "function cdb_route_with_waypoints(geometry, geometry, text, text[]) does not exist"]) + assert_in(e.message[0], ["Routing permission denied", "function cdb_route_with_waypoints(geometry, geometry, text, text[]) does not exist"])