def test_success_with_defaults(self):
geojson_to_spatialite(self.tmp.name, "tests/fixtures/geojson/valid.geojson")
# ensure all the right data was inserted
records = self.conn.execute(
"SELECT id, prop0, prop1, AsText(geometry) as geomtext FROM valid ORDER BY id;"
).fetchall()
self.assertEqual(3, len(records))
self.assertEqual((1, "string", None, "POINT(102 0.5)"), records[0])
self.assertEqual(
(2, "string", 0, "LINESTRING(102 0, 103 1, 104 0, 105 1)"), records[1]
)
self.assertEqual(
(3, "string", 7, "POLYGON((100 0, 101 0, 101 1, 100 1, 100 0))"), records[2]
)
# make sure the columns have the corect types
cols = self.conn.execute("PRAGMA table_info('valid');").fetchall()
self.assertDictEqual(
{
"id": "INTEGER",
"prop0": "TEXT",
"prop1": "FLOAT",
"geometry": "GEOMETRY",
},
{col[1]: col[2] for col in cols},
)
# ensure the spatial index was created
indexes = self.conn.execute(
"SELECT name FROM sqlite_master WHERE type='table' AND name='idx_valid_geometry';"
).fetchall()
self.assertEqual(1, len(indexes))
def test_failure_invalid_write_mode(self):
with self.assertRaises(ValueError):
geojson_to_spatialite(
self.tmp.name,
"tests/fixtures/geojson/valid.geojson",
write_mode="foobar",
)
def test_success_append_to_table(self):
geojson_to_spatialite(self.tmp.name, "tests/fixtures/geojson/valid.geojson")
geojson_to_spatialite(
self.tmp.name, "tests/fixtures/geojson/valid.geojson", write_mode="append"
)
records = self.conn.execute("SELECT * FROM valid ORDER BY id;").fetchall()
self.assertEqual(6, len(records))
def test_success_with_srid(self):
geojson_to_spatialite(
self.tmp.name, "tests/fixtures/geojson/valid.geojson", srid=27700
)
self.assertEqual(3, len(self.conn.execute("SELECT * FROM valid;").fetchall()))
self.assertEqual(
27700, self.conn.execute("SELECT srid(geometry) FROM valid;").fetchone()[0]
)
def test_failure_cant_append_to_table(self):
self.conn.execute("CREATE TABLE valid (id INT);")
with self.assertRaises(DataImportError):
geojson_to_spatialite(
self.tmp.name,
"tests/fixtures/geojson/valid.geojson",
write_mode="append",
)
def test_success_with_table_name(self):
geojson_to_spatialite(
self.tmp.name, "tests/fixtures/geojson/valid.geojson", table_name="foobar"
)
self.assertEqual(3, len(self.conn.execute("SELECT * FROM foobar;").fetchall()))
tables = self.conn.execute(
"SELECT name FROM sqlite_master WHERE type='table' AND name='valid';"
).fetchall()
self.assertEqual(0, len(tables))
def test_success_with_composite_key(self):
geojson_to_spatialite(
self.tmp.name, "tests/fixtures/geojson/valid.geojson", pk=["id", "prop0"]
)
self.assertEqual(3, len(self.conn.execute("SELECT * FROM valid;").fetchall()))
cols = self.conn.execute("PRAGMA table_info('valid');").fetchall()
self.assertDictEqual(
{"id": 1, "prop0": 2, "prop1": 0, "geometry": 0},
{col[1]: col[5] for col in cols},
)
def test_success_long_decimals(self):
geojson_to_spatialite(
self.tmp.name, "tests/fixtures/geojson/longcoords.geojson"
)
# for this test, we need to use KML for the output format
# because AsText applies 6 decimal places of rounding
# and we're trying to test that the full input precision is stored
records = self.conn.execute(
"SELECT id, prop0, AsKml(geometry) as geomtext FROM longcoords ORDER BY id;"
).fetchall()
self.assertEqual(1, len(records))
self.assertEqual(
(
1,
"string",
"<Point><coordinates>102.123456789,0.987654321</coordinates></Point>",
),
records[0],
)
def test_failure_pk_not_in_every_feature(self):
with self.assertRaises(DataImportError):
geojson_to_spatialite(
self.tmp.name, "tests/fixtures/geojson/valid.geojson", pk="prop1"
)
def test_failure_geojson_not_featurecollection(self):
with self.assertRaises(DataImportError):
geojson_to_spatialite(
self.tmp.name, "tests/fixtures/geojson/feature.geojson"
)
def test_failure_invalid_pk(self):
with self.assertRaises(TypeError):
geojson_to_spatialite(
self.tmp.name, "tests/fixtures/geojson/valid.geojson", pk=7
)
def test_failure_table_already_exists(self):
geojson_to_spatialite(self.tmp.name, "tests/fixtures/geojson/valid.geojson")
with self.assertRaises(DataImportError):
geojson_to_spatialite(self.tmp.name, "tests/fixtures/geojson/valid.geojson")
