In [1]:
import numpy as np
from pyCoCo import pyCoCo

In [2]:
fltPath = b"/Users/szymon/Projects/CoCo/data/filters"
rootPath = b"/Users/szymon/Projects/CoCo"

In [3]:
%%timeit 

coco = pyCoCo(fltPath, rootPath)


332 ms ± 877 µs per loop (mean ± std. dev. of 7 runs, 1 loop each)

In [3]:
coco = pyCoCo(fltPath, rootPath)

In [4]:
mjd = np.arange(55000, 55100, 5, dtype=float)
flt = [b"SDSS_r", b"SDSS_r", b"SDSS_r", b"SDSS_r", b"SDSS_r",
       b"SDSS_r", b"SDSS_r", b"SDSS_r", b"SDSS_r", b"SDSS_r",
       b"SDSS_i", b"SDSS_i", b"SDSS_i", b"SDSS_i", b"SDSS_i",
       b"SDSS_i", b"SDSS_i", b"SDSS_i", b"SDSS_i", b"SDSS_i"]
flt = np.array(flt)

In [5]:
%%timeit

coco.simulate(b"SN1994I", 0.1, -1.0, 0.2, 0.1, 3.1, 55000.0, mjd, flt)


164 ms ± 1.39 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

In [6]:
coco.simulate(b"SN1994I", 0.1, -1.0, 0.2, 0.1, 3.1, 55000.0, mjd, flt)


Out[6]:
array([[  1.03913696e-17,   7.78887051e-18,   4.73162858e-18,
          2.75936625e-18,   1.55442893e-18,   8.39496816e-19,
          4.30383900e-19,   2.07823934e-19,   9.57459112e-20,
          4.64839086e-20,   3.01981861e-21,   9.71810384e-20,
          3.28170991e-19,   6.04394911e-19,   8.31771215e-19,
          9.54611306e-19,   9.67843514e-19,   8.99224068e-19,
          7.84933257e-19,   6.54823219e-19],
       [  0.00000000e+00,   0.00000000e+00,   0.00000000e+00,
          0.00000000e+00,   0.00000000e+00,   0.00000000e+00,
          0.00000000e+00,   0.00000000e+00,   0.00000000e+00,
          0.00000000e+00,   0.00000000e+00,   0.00000000e+00,
          0.00000000e+00,   0.00000000e+00,   0.00000000e+00,
          0.00000000e+00,   0.00000000e+00,   0.00000000e+00,
          0.00000000e+00,   0.00000000e+00]])

In [8]:
paramMap = coco.get_fit_params()

In [10]:
paramMap.keys()


Out[10]:
dict_keys([b'SDSS_r', b'SDSS_i'])

In [ ]: