diff --git a/tests/test_lottery_predictor_collect.py b/tests/test_lottery_predictor_collect.py
new file mode 100644
index 0000000..010ba95
--- /dev/null
+++ b/tests/test_lottery_predictor_collect.py
@@ -0,0 +1,9 @@
+import pytest
+import lottery_predictor.collect as collect
+
+
+def test_get_record_count() -> None:
+    assert isinstance(collect.get_record_count('PowerballDraw'), int)
+    assert isinstance(collect.get_record_count('MegaMillionsDraw'), int)
+    with pytest.raises(ValueError):
+        collect.get_record_count('SomeNonExistentTable')