示例#1
0
class TestModel(unittest.TestCase):
    def setUp(self):
        self.postgresql = testing.postgresql.Postgresql()
        self.database = Database(url=self.postgresql.url())
        self.database.engine.execute('create extension postgis;')
        self.database.engine.execute('create extension postgis_topology;')
        self.database.create()
        self.root_dir = os.path.dirname(__file__)
        file_location = os.path.join(self.root_dir, 'data/sample-feed.zip')
        md5 = hashlib.md5(open(file_location, 'rb').read()).hexdigest()
        feed_file=FeedFile(md5sum=md5, file_url=file_location)
        database_load_versioned(feed_file, self.postgresql.url())

    def tearDown(self):
        self.database.session_factory.close_all()
        self.postgresql.stop()

    def test_agency_delete(self):
        session = self.database.get_session()
        agency = session.query(Agency).first()
        self.assertTrue(agency)
        session.delete(agency)
        session.commit()
        for table_cls in self.database.classes:
            if table_cls == FeedFile:
                continue
            self.assertEqual(0, session.query(table_cls).count(),
                             "Failed on table {}".format(table_cls))

    def test_create_geoms(self):
        session = self.database.get_session()
        shape_list = ShapeGeom.get_shape_list(session)
        for shape_id in shape_list:
            create_shapes_geom(self.database.url, shape_id[0])

    def test_import_feed(self):
        session = self.database.get_session()
        self.assertEqual(1, session.query(Agency).count())
        self.assertEqual(2, session.query(Calendar).count())
        self.assertEqual(1, session.query(CalendarDate).count())
        #self.assertEqual(2, session.query(FareAttribute).count())
        #self.assertEqual(4, session.query(FareRule).count())
        self.assertEqual(11, session.query(Frequency).count())
        self.assertEqual(5, session.query(Route).count())
        self.assertEqual(223, session.query(Shape).count())
        self.assertEqual(28, session.query(StopTime).count())
        self.assertEqual(9, session.query(Stop).count())
        self.assertEqual(11, session.query(Trip).count())

        for lon, lat, x, y in session.query(Stop.stop_lon, Stop.stop_lat, ST_X(Stop.the_geom), ST_Y(Stop.the_geom)).all():
            self.assertEqual(float(lon), x)
            self.assertEqual(float(lat), y)

        for lon, lat, x, y in session.query(Shape.shape_pt_lon, Shape.shape_pt_lat, ST_X(Shape.the_geom), ST_Y(Shape.the_geom)).all():
            self.assertAlmostEqual(float(lon), x)
            self.assertAlmostEqual(float(lat), y)

    def test_create_and_remove_index(self):
        self.database.drop_indexes()
        self.database.create_indexes()
示例#2
0
文件: upload.py 项目: Censio/gtfsdb
def main(database, parallel=0):

    db = Database(url=database, is_geospatial=True)
    db.create()

    feeds = set(get_gtfs_feeds(db.get_session()))

    db.drop_indexes()

    if parallel:
        concurrent_run(feeds, database, parallel)
    else:
        serial_run(feeds, database)

    create_shapes_geoms(db_url=database)

    db.create_indexes()