|
|
"""Test the pyproject.toml file for consistency and its dependencies.""" |
|
|
|
|
|
import glob |
|
|
import os |
|
|
from pathlib import Path |
|
|
|
|
|
from tomlkit import load |
|
|
|
|
|
ROOT_DIR = Path(__file__).parent.parent |
|
|
|
|
|
|
|
|
def test_optional_packages(): |
|
|
"""Ensure only required extensions are built and versions respect pyproject.toml""" |
|
|
with open(ROOT_DIR / "pyproject.toml") as f: |
|
|
data = load(f) |
|
|
dependencies = data["tool"]["poetry"]["dependencies"] |
|
|
extras = data["tool"]["poetry"]["extras"] |
|
|
all_packages = extras["all"] |
|
|
|
|
|
default_packages = [] |
|
|
optional_packages = [] |
|
|
|
|
|
for package, details in dependencies.items(): |
|
|
if isinstance(details, dict) and details.get("optional") is True: |
|
|
optional_packages.append(package) |
|
|
else: |
|
|
default_packages.append(package) |
|
|
|
|
|
|
|
|
assert sorted(optional_packages) == sorted(all_packages) |
|
|
assert sorted(optional_packages) == sorted(extras["all"]) |
|
|
|
|
|
|
|
|
assert set(default_packages).isdisjoint(set(optional_packages)) |
|
|
|
|
|
|
|
|
def test_default_package_files(): |
|
|
"""Ensure only required extensions are built and versions respect pyproject.toml""" |
|
|
with open(ROOT_DIR / "pyproject.toml") as f: |
|
|
data = load(f) |
|
|
dependencies = data["tool"]["poetry"]["dependencies"] |
|
|
package_files = glob.glob("openbb_platform/openbb/package/*.py") |
|
|
|
|
|
invalid_packages = [] |
|
|
default_packages = [] |
|
|
|
|
|
for package, details in dependencies.items(): |
|
|
if isinstance(details, dict) is False: |
|
|
default_packages.append(package) |
|
|
|
|
|
for file_path in package_files: |
|
|
package_name = os.path.basename(file_path).replace(".py", "") |
|
|
if package_name.startswith("_"): |
|
|
continue |
|
|
if "_" in package_name: |
|
|
base_package = package_name.split("_")[0] |
|
|
if "openbb-" + base_package not in default_packages: |
|
|
invalid_packages.append(package_name) |
|
|
elif "openbb-" + package_name not in default_packages: |
|
|
invalid_packages.append(package_name) |
|
|
|
|
|
assert not invalid_packages, ( |
|
|
f"If not making a PR, ignore this error -> " |
|
|
f"Found non-required extension static assets: {invalid_packages}. " |
|
|
f"Only required packages should be committed." |
|
|
f"Please create a new environment with only required extensions, " |
|
|
f"rebuild the static assets, and commit the changes." |
|
|
) |
|
|
|
