diff --git a/tests/test_train_test_split.py b/tests/test_train_test_split.py
new file mode 100644
index 0000000..7aac0e9
--- /dev/null
+++ b/tests/test_train_test_split.py
@@ -0,0 +1,34 @@
+import os
+
+import geopandas as gpd
+import numpy as np
+
+from elapid import train_test_split
+
+# set the test raster data paths
+directory_path, script_path = os.path.split(os.path.abspath(__file__))
+data_path = os.path.join(directory_path, "data")
+points = gpd.read_file(os.path.join(data_path, "test-point-samples.gpkg"))
+
+
+def test_checkerboard_split():
+    train, test = train_test_split.checkerboard_split(points, grid_size=1000)
+    assert isinstance(train, gpd.GeoDataFrame)
+
+    buffer = 500
+    xmin, ymin, xmax, ymax = points.total_bounds
+    buffered_bounds = [xmin - buffer, ymin - buffer, xmax + buffer, ymax + buffer]
+    train_buffered, test_buffered = train_test_split.checkerboard_split(points, grid_size=1000, bounds=buffered_bounds)
+    assert len(train_buffered) > len(train)
+
+
+def test_GeographicKFold():
+    n_folds = 4
+    gfolds = train_test_split.GeographicKFold(n_splits=n_folds)
+    counted_folds = 0
+    for train_idx, test_idx in gfolds.split(points):
+        train = points.iloc[train_idx]
+        test = points.iloc[test_idx]
+        assert len(train) > len(test)
+        counted_folds += 1
+    assert gfolds.get_n_splits() == n_folds == counted_folds