<commit_before>from unittest import TestCase
from nose.tools import assert_raises
from nose.tools import assert_not_equal, assert_equal
from ..helpers.integration_test_helper import IntegrationTestHelper


class TestDataObservatoryFunctions(TestCase):

    def setUp(self):
        self.env_variables = IntegrationTestHelper.get_environment_variables()
        self.sql_api_url = "https://{0}.{1}/api/v1/sql".format(
            self.env_variables['username'],
            self.env_variables['host'],
            self.env_variables['api_key']
        )

    def test_if_get_demographic_snapshot_is_ok(self):
        query = "SELECT duration, length, shape as the_geom " \
                "FROM obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669))".format(
                    self.env_variables['api_key'])
        routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
        assert_not_equal(routing['the_geom'], None)

    def test_if_get_demographic_snapshot_without_api_key_raise_error(self):
        query = "SELECT duration, length, shape as the_geom " \
                "FROM obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669))"
        try:
            IntegrationTestHelper.execute_query(self.sql_api_url, query)
        except Exception as e:
            assert_equal(e.message[0], "The api_key must be provided")

    def test_if_get_segment_snapshot_is_ok(self):
        query = "SELECT duration, length, shape as the_geom " \
                "FROM obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669))".format(
                    self.env_variables['api_key'])
        routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
        assert_not_equal(routing['the_geom'], None)

    def test_if_get_segment_snapshot_without_api_key_raise_error(self):
        query = "SELECT duration, length, shape as the_geom " \
                "FROM obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669))"
        try:
            IntegrationTestHelper.execute_query(self.sql_api_url, query)
        except Exception as e:
            assert_equal(e.message[0], "The api_key must be provided")<commit_msg>More fixes to data observatory integration tests<commit_after>from unittest import TestCase
from nose.tools import assert_raises
from nose.tools import assert_not_equal, assert_equal
from ..helpers.integration_test_helper import IntegrationTestHelper


class TestDataObservatoryFunctions(TestCase):

    def setUp(self):
        self.env_variables = IntegrationTestHelper.get_environment_variables()
        self.sql_api_url = "https://{0}.{1}/api/v1/sql".format(
            self.env_variables['username'],
            self.env_variables['host'],
            self.env_variables['api_key']
        )

    def test_if_get_demographic_snapshot_is_ok(self):
        query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669))".format(self.env_variables['api_key'])
        routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
        assert_not_equal(routing['the_geom'], None)

    def test_if_get_demographic_snapshot_without_api_key_raise_error(self):
        query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669))"
        try:
            IntegrationTestHelper.execute_query(self.sql_api_url, query)
        except Exception as e:
            assert_equal(e.message[0], "The api_key must be provided")

    def test_if_get_segment_snapshot_is_ok(self):
        query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669))".format(self.env_variables['api_key'])
        routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
        assert_not_equal(routing['the_geom'], None)

    def test_if_get_segment_snapshot_without_api_key_raise_error(self):
        query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669))"
        try:
            IntegrationTestHelper.execute_query(self.sql_api_url, query)
        except Exception as e:
            assert_equal(e.message[0], "The api_key must be provided")
{'fraction_non_alphanumeric': 0.09439960870628515, 'fraction_numerical': 0.03374908290535583, 'mean_word_length': 3.590347923681257, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 2, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '974158', 'n_tokens_mistral': 1359, 'n_tokens_neox': 1287, 'n_words': 207}