Setup and notebook to process size data

conda/diffaux_env.yml

@@ -0,0 +1,19 @@
+name: diffaux
+dependencies:
+    - python=3.11
+    - numpy 
+    - numba 
+    - flake8
+    - pytest
+    - jax
+    - mpi4py
+    - sphinx
+    - nbsphinx
+    - ipython
+    - jupyter 
+    - matplotlib
+    - scipy
+    - h5py
+    - pip
+    - astropy
+    - healpy

conda/env_requirements.txt

@@ -0,0 +1,140 @@
+alabaster==0.7.13
+anyio==4.4.0
+appnope==0.1.4
+argon2-cffi==23.1.0
+argon2-cffi-bindings==21.2.0
+arrow==1.3.0
+astroid==3.2.4
+asttokens==2.4.1
+async-lru==2.0.4
+attrs==23.2.0
+Babel==2.15.0
+backcall==0.2.0
+beautifulsoup4==4.12.3
+bleach==6.1.0
+certifi==2024.7.4
+cffi==1.16.0
+charset-normalizer==3.3.2
+comm==0.2.2
+contourpy==1.1.1
+cycler==0.12.1
+debugpy==1.8.2
+decorator==5.1.1
+defusedxml==0.7.1
+distlib==0.3.8
+docutils==0.20.1
+exceptiongroup==1.2.2
+executing==2.0.1
+fastjsonschema==2.20.0
+filelock==3.15.4
+fonttools==4.53.1
+fqdn==1.5.1
+h11==0.14.0
+h5py==3.11.0
+httpcore==1.0.5
+httpx==0.27.0
+idna==3.7
+imagesize==1.4.1
+importlib_metadata==8.2.0
+importlib_resources==6.4.0
+ipykernel==6.29.5
+ipython==8.12.3
+ipywidgets==8.1.3
+isoduration==20.11.0
+jedi==0.19.1
+Jinja2==3.1.4
+json5==0.9.25
+jsonpointer==3.0.0
+jsonschema==4.23.0
+jsonschema-specifications==2023.12.1
+jupyter==1.0.0
+jupyter-console==6.6.3
+jupyter-events==0.10.0
+jupyter-lsp==2.2.5
+jupyter_client==8.6.2
+jupyter_core==5.7.2
+jupyter_server==2.14.2
+jupyter_server_terminals==0.5.3
+jupyterlab==4.2.4
+jupyterlab_pygments==0.3.0
+jupyterlab_server==2.27.3
+jupyterlab_widgets==3.0.11
+jupytext==1.16.3
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.7.5
+matplotlib-inline==0.1.7
+mdit-py-plugins==0.4.1
+mdurl==0.1.2
+mistune==3.0.2
+nbclient==0.10.0
+nbconvert==7.16.4
+nbformat==5.10.4
+nbsphinx==0.9.4
+nest-asyncio==1.6.0
+notebook==7.2.1
+notebook_shim==0.2.4
+numpy==1.24.4
+overrides==7.7.0
+packaging==24.1
+pandocfilters==1.5.1
+parso==0.8.4
+pexpect==4.9.0
+pickleshare==0.7.5
+pillow==10.4.0
+pkgutil_resolve_name==1.3.10
+platformdirs==4.2.2
+prometheus_client==0.20.0
+prompt_toolkit==3.0.47
+psutil==6.0.0
+ptyprocess==0.7.0
+pure_eval==0.2.3
+pycparser==2.22
+Pygments==2.18.0
+pyparsing==3.1.2
+python-dateutil==2.9.0.post0
+python-json-logger==2.0.7
+pytz==2024.1
+PyYAML==6.0.1
+pyzmq==26.0.3
+qtconsole==5.5.2
+QtPy==2.4.1
+referencing==0.35.1
+requests==2.32.3
+rfc3339-validator==0.1.4
+rfc3986-validator==0.1.1
+rpds-py==0.19.1
+Send2Trash==1.8.3
+six==1.16.0
+sniffio==1.3.1
+snowballstemmer==2.2.0
+soupsieve==2.5
+Sphinx==7.1.2
+sphinx-autoapi==3.2.1
+sphinx-copybutton==0.5.2
+sphinx-rtd-theme==2.0.0
+sphinxcontrib-applehelp==1.0.4
+sphinxcontrib-devhelp==1.0.2
+sphinxcontrib-htmlhelp==2.0.1
+sphinxcontrib-jquery==4.1
+sphinxcontrib-jsmath==1.0.1
+sphinxcontrib-qthelp==1.0.3
+sphinxcontrib-serializinghtml==1.1.5
+stack-data==0.6.3
+terminado==0.18.1
+tinycss2==1.3.0
+tomli==2.0.1
+tornado==6.4.1
+traitlets==5.14.3
+types-python-dateutil==2.9.0.20240316
+typing_extensions==4.12.2
+uri-template==1.3.0
+urllib3==2.2.2
+virtualenv==20.26.3
+wcwidth==0.2.13
+webcolors==24.6.0
+webencodings==0.5.1
+websocket-client==1.8.0
+widgetsnbextension==4.0.11
+zipp==3.19.2

src/diffaux/size_modeling/tests/test_zhang_yang17.py

@@ -0,0 +1,25 @@
+"""
+"""
+import numpy as np
+
+from ..zhang_yang17 import (
+    mc_size_vs_luminosity_early_type,
+    mc_size_vs_luminosity_late_type,
+)
+
+NGALS_TEST = int(1e7)
+SIZE_MAX = 101.0  # kpc
+
+
+def test_mc_size_vs_luminosity_early_type():
+    magr = np.random.uniform(-100, 100, NGALS_TEST)
+    redshift = np.random.uniform(0, 20, NGALS_TEST)
+    size = mc_size_vs_luminosity_early_type(magr, redshift)
+    assert np.all(size < SIZE_MAX)
+
+
+def test_mc_size_vs_luminosity_late_type():
+    magr = np.random.uniform(-100, 100, NGALS_TEST)
+    redshift = np.random.uniform(0, 20, NGALS_TEST)
+    size = mc_size_vs_luminosity_late_type(magr, redshift)
+    assert np.all(size < SIZE_MAX)