diffsky 0.1.1__tar.gz → 0.2.0__tar.gz

diffsky-0.2.0/.github/dependabot.yml

@@ -0,0 +1,10 @@
+version: 2
+updates:
+  - package-ecosystem: "github-actions"
+    directory: "/"
+    schedule:
+      interval: "monthly"
+    groups:
+      github-actions:
+        patterns:
+          - '*'

diffsky-0.1.1/.github/workflows/linting.yml → diffsky-0.2.0/.github/workflows/linting.yaml

@@ -12,25 +12,25 @@ jobs:
     runs-on: "ubuntu-latest"
 
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v4
 
-      - uses: conda-incubator/setup-miniconda@v2
+      - uses: conda-incubator/setup-miniconda@v3
         with:
-          python-version: 3.9
+          python-version: 3.11
           channels: conda-forge,defaults
           channel-priority: strict
           show-channel-urls: true
           miniforge-version: latest
-          miniforge-variant: Mambaforge
 
       - name: configure conda and install code
         shell: bash -l {0}
         run: |
-          mamba install --quiet \
+          conda install --quiet \
             --file=requirements.txt
           python -m pip install --no-deps -e .
-          mamba install -y -q \
-            flake8
+          conda install -y -q \
+            flake8 \
+            flake8-pyproject
 
       - name: lint
         shell: bash -l {0}

diffsky-0.2.0/.github/workflows/monthly-warning-test.yaml

@@ -0,0 +1,64 @@
+name: Test for Warnings
+
+on:
+  workflow_dispatch: null
+  schedule:
+    # Runs "First of every month at 3:15am Central"
+    - cron: '15 8 1 * *'
+
+jobs:
+  tests:
+    name: tests
+    runs-on: "ubuntu-latest"
+
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - uses: conda-incubator/setup-miniconda@v3
+        with:
+          python-version: 3.11
+          channels: conda-forge,defaults
+          channel-priority: strict
+          show-channel-urls: true
+          miniforge-version: latest
+
+
+      - name: configure conda and install code
+      # Test against current main branch of all diff+ dependencies
+        shell: bash -l {0}
+        env:
+          GITHUB_TOKEN: ${{ secrets.CPAC_PRIVATE_REPO_TOKEN }}
+          GITHUB_USER: ${{ secrets.GH_USR_TESTING_TOKEN }}
+
+      # Test against current main branch of all diff+ dependencies
+        run: |
+          conda config --set always_yes yes
+          conda install --quiet \
+            --file=requirements.txt
+          conda install -y -q \
+            flake8 \
+            pytest \
+            pytest-xdist \
+            pytest-cov \
+            pip \
+            setuptools \
+            "setuptools_scm>=7,<8" \
+            matplotlib \
+            astropy \
+            python-build
+          pip uninstall diffmah --yes
+          pip uninstall diffstar --yes
+          pip uninstall dsps --yes
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffmah.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffstar.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/dsps.git
+          pip install --no-deps git+https://${GITHUB_USER}:${GITHUB_TOKEN}@github.com/ArgonneCPAC/diffstarpop.git
+          python -m pip install --no-build-isolation --no-deps -e .
+
+      - name: test that no warnings are raised
+        shell: bash -l {0}
+        run: |
+          export PYTHONWARNINGS=error
+          pytest -v diffsky --cov --cov-report=xml

diffsky-0.1.1/.github/workflows/tests.yml → diffsky-0.2.0/.github/workflows/test_latest_releases.yaml

@@ -1,4 +1,4 @@
-name: tests
+name: test-latest-diffstuff-releases
 
 on:
   push:
@@ -8,31 +8,35 @@ on:
 
 jobs:
   tests:
-    name: tests
+    name: test-latest-diffstuff-releases
     runs-on: "ubuntu-latest"
 
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v4
         with:
           fetch-depth: 0
 
-      - uses: conda-incubator/setup-miniconda@v2
+      - uses: conda-incubator/setup-miniconda@v3
         with:
-          python-version: 3.9
+          python-version: 3.11
           channels: conda-forge,defaults
           channel-priority: strict
           show-channel-urls: true
           miniforge-version: latest
-          miniforge-variant: Mambaforge
-          use-mamba: true
 
       - name: configure conda and install code
+      # Test against current main branch of all diff+ dependencies
         shell: bash -l {0}
+        env:
+          GITHUB_TOKEN: ${{ secrets.CPAC_PRIVATE_REPO_TOKEN }}
+          GITHUB_USER: ${{ secrets.GH_USR_TESTING_TOKEN }}
+
+      # Test against conda-forge for all dependencies except use main branch for dsps
         run: |
           conda config --set always_yes yes
-          mamba install --quiet \
+          conda install --quiet \
             --file=requirements.txt
-          mamba install -y -q \
+          conda install -y -q \
             flake8 \
             pytest \
             pytest-xdist \
@@ -40,7 +44,12 @@ jobs:
             pip \
             setuptools \
             "setuptools_scm>=7,<8" \
+            matplotlib \
+            astropy \
             python-build
+          pip uninstall dsps --yes
+          pip install --no-deps git+https://github.com/ArgonneCPAC/dsps.git
+          pip install --no-deps git+https://${GITHUB_USER}:${GITHUB_TOKEN}@github.com/ArgonneCPAC/diffstarpop.git
           python -m pip install --no-build-isolation --no-deps -e .
 
       - name: test
@@ -49,7 +58,9 @@ jobs:
           pytest -v diffsky --cov --cov-report=xml
 
       - name: Upload coverage reports to Codecov
-        uses: codecov/codecov-action@v3
+        uses: codecov/codecov-action@v5
+        env:
+          CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
 
       - name: test versions
         shell: bash -el {0}

diffsky-0.2.0/.github/workflows/test_main_branch.yaml

@@ -0,0 +1,91 @@
+name: test-main-branch-diffstuff
+
+on:
+  push:
+    branches:
+      - main
+  pull_request: null
+
+jobs:
+  tests:
+    name: test-main-branch-diffstuff
+    runs-on: "ubuntu-latest"
+
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - uses: conda-incubator/setup-miniconda@v3
+        with:
+          python-version: 3.11
+          channels: conda-forge,defaults
+          channel-priority: strict
+          show-channel-urls: true
+          miniforge-version: latest
+
+      - name: configure conda and install code
+      # Test against current main branch of all diff+ dependencies
+        shell: bash -l {0}
+        env:
+          GITHUB_TOKEN: ${{ secrets.CPAC_PRIVATE_REPO_TOKEN }}
+          GITHUB_USER: ${{ secrets.GH_USR_TESTING_TOKEN }}
+
+      # Test against conda-forge for all dependencies except use main branch for dsps
+        run: |
+          conda config --set always_yes yes
+          conda install --quiet \
+            --file=requirements.txt
+          conda install -y -q \
+            flake8 \
+            pytest \
+            pytest-xdist \
+            pytest-cov \
+            pip \
+            setuptools \
+            "setuptools_scm>=7,<8" \
+            matplotlib \
+            astropy \
+            python-build
+          pip uninstall diffmah --yes
+          pip uninstall diffstar --yes
+          pip uninstall dsps --yes
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffmah.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffstar.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/dsps.git
+          pip install --no-deps git+https://${GITHUB_USER}:${GITHUB_TOKEN}@github.com/ArgonneCPAC/diffstarpop.git
+          python -m pip install --no-build-isolation --no-deps -e .
+
+      - name: test
+        shell: bash -l {0}
+        run: |
+          pytest -v diffsky --cov --cov-report=xml
+
+      - name: Upload coverage reports to Codecov
+        uses: codecov/codecov-action@v5
+        env:
+          CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
+
+      - name: test versions
+        shell: bash -el {0}
+        run: |
+          pip uninstall diffsky --yes
+          [[ $(python setup.py --version) != "0.0.0" ]] || exit 1
+
+          rm -rf dist/*
+          python setup.py sdist
+          pip install --no-deps --no-build-isolation dist/*.tar.gz
+          pushd ..
+          python -c "import diffsky; assert diffsky.__version__ != '0.0.0'"
+          popd
+          pip uninstall diffsky --yes
+
+          rm -rf dist/*
+          python -m build --sdist . --outdir dist
+          pip install --no-deps --no-build-isolation dist/*.tar.gz
+          pushd ..
+          python -c "import diffsky; assert diffsky.__version__ != '0.0.0'"
+          popd
+          pip uninstall diffsky --yes
+
+          python -m pip install -v --no-deps --no-build-isolation -e .

diffsky-0.2.0/.github/workflows/tests_cron.yaml

@@ -0,0 +1,95 @@
+name: test-main-branch-diffstuff
+
+on:
+  workflow_dispatch: null
+  schedule:
+    # Runs "every Monday & Thursday at 3:05am Central"
+    - cron: '5 8 * * 1,4'
+  push:
+    branches:
+      - main
+  pull_request: null
+
+jobs:
+  tests:
+    name: tests
+    runs-on: "ubuntu-latest"
+
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - uses: conda-incubator/setup-miniconda@v3
+        with:
+          python-version: 3.11
+          channels: conda-forge,defaults
+          channel-priority: strict
+          show-channel-urls: true
+          miniforge-version: latest
+
+      - name: configure conda and install code
+      # Test against current main branch of all diff+ dependencies
+        shell: bash -l {0}
+        env:
+          GITHUB_TOKEN: ${{ secrets.CPAC_PRIVATE_REPO_TOKEN }}
+          GITHUB_USER: ${{ secrets.GH_USR_TESTING_TOKEN }}
+
+      # Test against current main branch of all diff+ dependencies
+        run: |
+          conda config --set always_yes yes
+          conda install --quiet \
+            --file=requirements.txt
+          conda install -y -q \
+            flake8 \
+            pytest \
+            pytest-xdist \
+            pytest-cov \
+            pip \
+            setuptools \
+            "setuptools_scm>=7,<8" \
+            matplotlib \
+            astropy \
+            python-build
+          pip uninstall diffmah --yes
+          pip uninstall diffstar --yes
+          pip uninstall dsps --yes
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffmah.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffstar.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/dsps.git
+          pip install --no-deps git+https://${GITHUB_USER}:${GITHUB_TOKEN}@github.com/ArgonneCPAC/diffstarpop.git
+          python -m pip install --no-build-isolation --no-deps -e .
+
+      - name: test
+        shell: bash -l {0}
+        run: |
+          pytest -v diffsky --cov --cov-report=xml
+
+      - name: Upload coverage reports to Codecov
+        uses: codecov/codecov-action@v5
+        env:
+          CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
+
+      - name: test versions
+        shell: bash -el {0}
+        run: |
+          pip uninstall diffsky --yes
+          [[ $(python setup.py --version) != "0.0.0" ]] || exit 1
+
+          rm -rf dist/*
+          python setup.py sdist
+          pip install --no-deps --no-build-isolation dist/*.tar.gz
+          pushd ..
+          python -c "import diffsky; assert diffsky.__version__ != '0.0.0'"
+          popd
+          pip uninstall diffsky --yes
+
+          rm -rf dist/*
+          python -m build --sdist . --outdir dist
+          pip install --no-deps --no-build-isolation dist/*.tar.gz
+          pushd ..
+          python -c "import diffsky; assert diffsky.__version__ != '0.0.0'"
+          popd
+          pip uninstall diffsky --yes
+
+          python -m pip install -v --no-deps --no-build-isolation -e .

{diffsky-0.1.1 → diffsky-0.2.0}/.readthedocs.yml

@@ -6,7 +6,7 @@ sphinx:
 build:
   os: "ubuntu-22.04"
   tools:
-    python: "mambaforge-22.9"
+    python: "miniconda-latest"
 
 conda:
   environment: docs/source/rtd_environment.yaml

diffsky-0.2.0/CHANGES.rst

@@ -0,0 +1,18 @@
+0.2.0 (2025-03-24)
+-------------------
+- Update population-level photometry models of dust and burstiness
+
+
+0.1.2 (2024-10-25)
+-------------------
+- Update calls to diffmah v0.6.1 and diffstar v0.3.2
+
+
+0.1.1 (2023-10-04)
+-------------------
+- Update calls to dsps v0.3.4
+
+
+0.1.0 (2023-10-03)
+-------------------
+- First release. Compatible with diffstar v0.2.1 and dsps v0.3.3.

{diffsky-0.1.1 → diffsky-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: diffsky
-Version: 0.1.1
+Version: 0.2.0
 Summary: Library for differentiable generation of synthetic skies
 Author-email: Andrew Hearin <ahearin@anl.gov>
 License: BSD 3-Clause License
@@ -35,15 +35,16 @@ License: BSD 3-Clause License
         
 Project-URL: home, https://github.com/ArgonneCPAC/diffsky
 Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.9
+Requires-Python: >=3.11
 Description-Content-Type: text/x-rst
 License-File: LICENSE.rst
 Requires-Dist: numpy
 Requires-Dist: jax
+Requires-Dist: diffmah>=0.7.0
+Requires-Dist: diffstar>=0.3.3
+Requires-Dist: dsps>=0.4.0
 Requires-Dist: h5py
-Requires-Dist: diffmah
-Requires-Dist: diffstar
-Requires-Dist: dsps
+Dynamic: license-file
 
 diffsky
 ============
@@ -69,7 +70,7 @@ Conda environment
 ~~~~~~~~~~~~~~~~~
 For a typical development environment in conda-forge::
 
-    $ conda create -c conda-forge -n diffsky_env python=3.9 numpy jax pytest ipython jupyter matplotlib scipy h5py diffmah diffstar dsps diffsky
+    $ conda create -c conda-forge -n diffsky_env python=3.11 numpy jax pytest ipython jupyter matplotlib scipy h5py diffmah diffstar dsps diffsky
 
 
 Documentation

{diffsky-0.1.1 → diffsky-0.2.0}/README.rst

@@ -22,7 +22,7 @@ Conda environment
 ~~~~~~~~~~~~~~~~~
 For a typical development environment in conda-forge::
 
-    $ conda create -c conda-forge -n diffsky_env python=3.9 numpy jax pytest ipython jupyter matplotlib scipy h5py diffmah diffstar dsps diffsky
+    $ conda create -c conda-forge -n diffsky_env python=3.11 numpy jax pytest ipython jupyter matplotlib scipy h5py diffmah diffstar dsps diffsky
 
 
 Documentation

{diffsky-0.1.1 → diffsky-0.2.0}/diffsky/__init__.py

@@ -1,5 +1,7 @@
 """
 """
+
 # flake8: noqa
 
 from ._version import __version__
+from .mass_functions.mc_diffmah_tpeak import mc_subhalos

diffsky-0.2.0/diffsky/_version.py

@@ -0,0 +1 @@
+__version__ = '0.2.0'

diffsky-0.2.0/diffsky/burstpop/__init__.py

@@ -0,0 +1,11 @@
+"""
+"""
+# flake8: noqa
+
+from .diffburstpop import (
+    DEFAULT_DIFFBURSTPOP_PARAMS,
+    DEFAULT_DIFFBURSTPOP_U_PARAMS,
+    calc_bursty_age_weights_from_diffburstpop_params,
+    get_bounded_diffburstpop_params,
+    get_unbounded_diffburstpop_params,
+)

diffsky-0.2.0/diffsky/burstpop/diagnostics/plot_fburstpop.py

@@ -0,0 +1,60 @@
+"""
+"""
+
+import numpy as np
+from matplotlib import pyplot as plt
+
+from ..fburstpop import DEFAULT_FBURSTPOP_PARAMS, get_lgfburst_from_fburstpop_params
+
+
+def make_fburstpop_comparison_plot(
+    params,
+    params2=DEFAULT_FBURSTPOP_PARAMS,
+    fname=None,
+    label1=r"${\rm new\ model}$",
+    label2=r"${\rm default\ model}$",
+):
+    """Make basic diagnostic plot of the model for Fburst
+
+    Parameters
+    ----------
+    params : namedtuple
+        Instance of fburstpop.FburstPopParams
+
+    params2 : namedtuple, optional
+        Instance of fburstpop.FburstPopParams
+        Default is set by DEFAULT_FBURSTPOP_PARAMS
+
+    fname : string, optional
+        filename of the output figure
+
+    """
+    nsm, nsfr = 250, 250
+    logsm_grid = np.linspace(7, 12, nsm)
+    logssfr_grid = np.linspace(-13, -8, nsfr)
+
+    X, Y = np.meshgrid(logsm_grid, logssfr_grid)
+
+    Z = get_lgfburst_from_fburstpop_params(params, X, Y)
+    Z2 = get_lgfburst_from_fburstpop_params(params2, X, Y)
+
+    fig, axes = plt.subplots(1, 2, figsize=(10, 4))
+    (ax0, ax1) = axes
+
+    pcm0 = ax0.pcolor(X, Y, Z, cmap="coolwarm_r", vmin=-4.5, vmax=-2.1)
+    fig.colorbar(pcm0, ax=ax0)
+
+    pcm1 = ax1.pcolor(X, Y, Z2, cmap="coolwarm_r", vmin=-4.5, vmax=-2.1)
+    fig.colorbar(pcm1, ax=ax1, label=r"${\rm lgFburst}$")
+    for ax in axes:
+        xlabel = ax.set_xlabel(r"$\log_{10}M_{\star}/M_{\odot}$")
+    ylabel = ax0.set_ylabel(r"${\rm \log_{10}sSFR}$")
+
+    ax0.set_title(label1)
+    ax1.set_title(label2)
+
+    if fname is not None:
+        fig.savefig(
+            fname, bbox_extra_artists=[xlabel, ylabel], bbox_inches="tight", dpi=200
+        )
+    return fig

diffsky-0.2.0/diffsky/burstpop/diagnostics/plot_tburstpop.py

@@ -0,0 +1,115 @@
+"""
+"""
+
+import numpy as np
+from dsps.sfh import diffburst
+from matplotlib import lines as mlines
+from matplotlib import pyplot as plt
+
+from .. import tburstpop as tbp
+
+
+def make_tburstpop_comparison_plot(
+    params,
+    params2=tbp.DEFAULT_TBURSTPOP_PARAMS,
+    fname=None,
+    label1=r"${\rm new\ model}$",
+    label2=r"${\rm default\ model}$",
+):
+    """Make basic diagnostic plot of the model for Tburst
+
+    Parameters
+    ----------
+    params : namedtuple
+        Instance of tburstpop.TburstPopParams
+
+    params2 : namedtuple, optional
+        Instance of tburstpop.TburstPopParams
+        Default is set by DEFAULT_TBURSTPOP_PARAMS
+
+    fname : string, optional
+        filename of the output figure
+
+    """
+    lgyrarr = np.linspace(5, 9.05, 100)
+
+    logsmarr = np.array((9.0, 9.0, 12.0, 12.0))
+    logssfrarr = np.array((-7.0, -13.0, -7.0, -13.0))
+    lgyr_peak, lgyr_max = tbp.get_tburst_params_from_tburstpop_params(
+        tbp.DEFAULT_TBURSTPOP_PARAMS, logsmarr, logssfrarr
+    )
+
+    u_params = np.array(tbp.DEFAULT_TBURSTPOP_U_PARAMS) + np.random.uniform(
+        -1, 1, len(tbp.DEFAULT_TBURSTPOP_PARAMS)
+    )
+    u_params = tbp.DEFAULT_TBURSTPOP_U_PARAMS._make(u_params)
+    alt_params = tbp.get_bounded_tburstpop_params(u_params)
+    lgyr_peak2, lgyr_max2 = tbp.get_tburst_params_from_tburstpop_params(
+        alt_params, logsmarr, logssfrarr
+    )
+
+    fig, (ax, ax1) = plt.subplots(1, 2, figsize=(12, 5), sharex=True, sharey=True)
+    ax.loglog()
+    ax.set_xlim(8e4, 2e9)
+    ax.set_ylim(2e-7, 10.5)
+    xlabel = ax.set_xlabel(r"$\tau_{\rm age}\ {\rm [yr]}$")
+    xlabel = ax1.set_xlabel(r"$\tau_{\rm age}\ {\rm [yr]}$")
+    ylabel = ax.set_ylabel(r"${\rm PDF}$")
+
+    age_weights = diffburst._pureburst_age_weights_from_params(
+        lgyrarr, lgyr_peak[0], lgyr_max[0]
+    )
+    ax.plot(10**lgyrarr, age_weights, color="blue")
+    age_weights2 = diffburst._pureburst_age_weights_from_params(
+        lgyrarr, lgyr_peak2[0], lgyr_max2[0]
+    )
+    ax.plot(10**lgyrarr, age_weights2, "--", color="blue")
+
+    age_weights = diffburst._pureburst_age_weights_from_params(
+        lgyrarr, lgyr_peak[1], lgyr_max[1]
+    )
+    ax.plot(10**lgyrarr, age_weights, color="red")
+    age_weights2 = diffburst._pureburst_age_weights_from_params(
+        lgyrarr, lgyr_peak2[1], lgyr_max2[1]
+    )
+    ax.plot(10**lgyrarr, age_weights2, "--", color="red")
+
+    age_weights = diffburst._pureburst_age_weights_from_params(
+        lgyrarr, lgyr_peak[2], lgyr_max[2]
+    )
+    ax1.plot(10**lgyrarr, age_weights, color="blue")
+    age_weights2 = diffburst._pureburst_age_weights_from_params(
+        lgyrarr, lgyr_peak2[2], lgyr_max2[2]
+    )
+    ax1.plot(10**lgyrarr, age_weights2, "--", color="blue")
+
+    age_weights = diffburst._pureburst_age_weights_from_params(
+        lgyrarr, lgyr_peak[3], lgyr_max[3]
+    )
+    ax1.plot(10**lgyrarr, age_weights, color="red")
+    age_weights2 = diffburst._pureburst_age_weights_from_params(
+        lgyrarr, lgyr_peak2[3], lgyr_max2[3]
+    )
+    ax1.plot(10**lgyrarr, age_weights2, "--", color="red")
+
+    ax.set_title(r"$M_{\star}=10^9M_{\odot}$")
+    ax1.set_title(r"$M_{\star}=10^{12}M_{\odot}$")
+
+    red_line = mlines.Line2D([], [], ls="-", c="red", label=r"${\rm Q}$")
+    blue_line = mlines.Line2D([], [], ls="-", c="blue", label=r"${\rm SF}$")
+    solid_line = mlines.Line2D(
+        [], [], ls="-", c="gray", label=r"${\rm default\ model}$"
+    )
+    dashed_line = mlines.Line2D([], [], ls="--", c="gray", label=r"${\rm new\ model}$")
+    leg0 = ax.legend(handles=[red_line, blue_line], loc="upper left")
+    ax.add_artist(leg0)
+    ax.legend(handles=[solid_line, dashed_line], loc="upper right")
+    leg0 = ax1.legend(handles=[red_line, blue_line], loc="upper left")
+    ax1.add_artist(leg0)
+    ax1.legend(handles=[solid_line, dashed_line], loc="upper right")
+
+    if fname is not None:
+        fig.savefig(
+            fname, bbox_extra_artists=[xlabel, ylabel], bbox_inches="tight", dpi=200
+        )
+    return fig

diffsky-0.2.0/diffsky/burstpop/diagnostics/tests/test_plot_fburstpop.py

@@ -0,0 +1,15 @@
+"""
+"""
+
+import os
+
+from ..plot_fburstpop import DEFAULT_FBURSTPOP_PARAMS, make_fburstpop_comparison_plot
+
+_THIS_DRNAME = os.path.dirname(os.path.abspath(__file__))
+
+
+def test_make_freqburst_comparison_plot():
+    fn = os.path.join(_THIS_DRNAME, "dummy.png")
+    make_fburstpop_comparison_plot(DEFAULT_FBURSTPOP_PARAMS, fname=fn)
+    assert os.path.isfile(fn)
+    os.remove(fn)

diffsky-0.2.0/diffsky/burstpop/diagnostics/tests/test_plot_tburstpop.py

@@ -0,0 +1,16 @@
+"""
+"""
+
+import os
+
+from ... import tburstpop as tbp
+from ..plot_tburstpop import make_tburstpop_comparison_plot
+
+_THIS_DRNAME = os.path.dirname(os.path.abspath(__file__))
+
+
+def test_make_freqburst_comparison_plot():
+    fn = os.path.join(_THIS_DRNAME, "dummy.png")
+    make_tburstpop_comparison_plot(tbp.DEFAULT_TBURSTPOP_PARAMS, fname=fn)
+    assert os.path.isfile(fn)
+    os.remove(fn)