Fix test and address comments

Author: lajohn4747

.github/workflows/readme.yml

@@ -22,5 +22,6 @@ jobs:
       run: |
           python -m pip install --upgrade pip
           python -m pip install invoke rundoc .
+          python -m pip install tomli
     - name: Run the README.md
       run: invoke readme

pyproject.toml

@@ -42,9 +42,7 @@ dependencies = [
 requires = ['setuptools', 'wheel']
 build-backend = "setuptools.build_meta"
 
-# Development dependencies
 [project.optional-dependencies]
-
 torch = [
     "torch>=1.8.0;python_version<'3.10'",
     "torch>=1.11.0;python_version>='3.10' and python_version<'3.11'",
@@ -60,6 +58,7 @@ test = [
     'pytest-rerunfailures>=10',
     'jupyter>=1.0.0,<2',
     'rundoc>=0.4.3,<0.5',
+    'tomli>=2.0.0,<3',
     'invoke',
 ]
 

tasks.py

@@ -2,7 +2,8 @@
 import inspect
 import operator
 import os
-import toml
+import tomli
+import sys
 from packaging.requirements import Requirement
 from packaging.version import Version
 import shutil
@@ -72,7 +73,7 @@ def _get_minimum_versions(dependencies, python_version):
 @task
 def install_minimum(c):
     with open('pyproject.toml', 'r', encoding='utf-8') as pyproject_file:
-        pyproject_data = toml.load(pyproject_file)
+        pyproject_data = tomli.load(pyproject_file)
 
     dependencies = pyproject_data.get('project', {}).get('dependencies', [])
     python_version = '.'.join(map(str, sys.version_info[:2]))

tests/test_tasks.py

@@ -0,0 +1,36 @@
+from tasks import _get_minimum_versions
+
+
+def test_get_minimum_versions():
+    """Test the ``_get_minimum_versions`` method.
+    The method should return the minimum versions of the dependencies for the given python version.
+    If a library is linked to an URL, the minimum version should be the URL.
+    """
+    # Setup
+    dependencies = [
+        "numpy>=1.20.0,<2;python_version<'3.10'",
+        "numpy>=1.23.3,<2;python_version>='3.10'",
+        "pandas>=1.2.0,<2;python_version<'3.10'",
+        "pandas>=1.3.0,<2;python_version>='3.10'",
+        'humanfriendly>=8.2,<11',
+        'pandas @ git+https://github.com/pandas-dev/pandas.git@master#egg=pandas'
+    ]
+
+    # Run
+    minimum_versions_39 = _get_minimum_versions(dependencies, '3.9')
+    minimum_versions_310 = _get_minimum_versions(dependencies, '3.10')
+
+    # Assert
+    expected_versions_39 = [
+        'numpy==1.20.0',
+        'pandas @ git+https://github.com/pandas-dev/pandas.git@master#egg=pandas',
+        'humanfriendly==8.2',
+    ]
+    expected_versions_310 = [
+        'numpy==1.23.3',
+        'pandas @ git+https://github.com/pandas-dev/pandas.git@master#egg=pandas',
+        'humanfriendly==8.2',
+    ]
+
+    assert minimum_versions_39 == expected_versions_39
+    assert minimum_versions_310 == expected_versions_310