diff --git a/src/gamedata/tests.rs b/src/gamedata/tests.rs
index 7b56817..e975823 100644
--- a/src/gamedata/tests.rs
+++ b/src/gamedata/tests.rs
@@ -1,4 +1,4 @@
-use crate::gamedata::score_checkers::Direction;
+use crate::gamedata::{heuristic::potential_wins, score_checkers::Direction};
 
 use super::*;
 
@@ -140,3 +140,14 @@ fn game_over_test() {
     board.columns.set(0, 0, Disk::EMPTY).expect("balls");
     assert!(!board.game_over());
 }
+
+#[test]
+fn heuristic_pot_wins() {
+    let mut board = Board::default();
+    board.play(Disk::BLU, 3);
+    board.play(Disk::BLU, 3);
+    board.play(Disk::BLU, 3);
+    board.play(Disk::BLU, 2);
+    board.play(Disk::BLU, 1);
+    assert_eq!(16, potential_wins(&board.columns, &Disk::BLU));
+}