From 22e09638b9ddfaaf283d1bf2fd78718d56f8de02 Mon Sep 17 00:00:00 2001
From: Ethan Henderson <ethan.henderson.1998@gmail.com>
Date: Mon, 27 Sep 2021 13:32:11 +0100
Subject: [PATCH] Ensure only the relevant versions are tested

---
 noxfile.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/noxfile.py b/noxfile.py
index a9b5677..23ab73f 100644
--- a/noxfile.py
+++ b/noxfile.py
@@ -1,11 +1,19 @@
+import os
 from pathlib import Path
 
 import nox
 
+if os.environ.get("CI", False):
+    import sys
+
+    PY_VERSIONS = [".".join(f"{v}" for v in sys.version_info[:2])]
+
+else:
+    PY_VERSIONS = [f"3.{v}" for v in range(6, 11)]  # 3.6 - 3.10
+
 PROJECT_NAME = "analytix"
 LIB_DIR = Path(__file__).parent / PROJECT_NAME
 TEST_DIR = Path(__file__).parent / "tests"
-PY_VERSIONS = [f"3.{v}" for v in range(6, 11)]  # 3.6 - 3.10
 
 
 def parse_requirements(path):