lsst.dax.apdb.tests.data_factory Namespace Reference

Functions
pandas.DataFrame	makeObjectCatalog (Region region, int count)
pandas.DataFrame	makeSourceCatalog (pandas.DataFrame objects, DateTime visit_time, int start_id=0, int ccdVisitId=1)
pandas.DataFrame	makeForcedSourceCatalog (pandas.DataFrame objects, DateTime visit_time, int ccdVisitId=1)

Function Documentation

makeForcedSourceCatalog()

pandas.DataFrame lsst.dax.apdb.tests.data_factory.makeForcedSourceCatalog	(	pandas.DataFrame	objects,
		DateTime	visit_time,
		int	ccdVisitId = 1
	)

Make a catalog containing a bunch of DiaFourceSources associated with
the input DiaObjects.

Parameters
----------
objects : `pandas.DataFrame`
    Catalog of DiaObject records.
visit_time : `lsst.daf.base.DateTime`
    Time of the visit.
ccdVisitId : `int`
    Value for ``ccdVisitId`` field.

Returns
-------
catalog : `pandas.DataFrame`
    Catalog of DiaForcedSource records.

Notes
-----
Returned catalog only contains small number of columns needed for tests.

Definition at line 136 of file data_factory.py.

                            ccdVisitId: int = 1) -> pandas.DataFrame:
    """Make a catalog containing a bunch of DiaFourceSources associated with
    the input DiaObjects.
 
    Parameters
    ----------
    objects : `pandas.DataFrame`
        Catalog of DiaObject records.
    visit_time : `lsst.daf.base.DateTime`
        Time of the visit.
    ccdVisitId : `int`
        Value for ``ccdVisitId`` field.
 
    Returns
    -------
    catalog : `pandas.DataFrame`
        Catalog of DiaForcedSource records.
 
    Notes
    -----
    Returned catalog only contains small number of columns needed for tests.
    """
    nrows = len(objects)
    midPointTai = visit_time.get(system=DateTime.MJD)
    df = pandas.DataFrame({
        "diaObjectId": objects["diaObjectId"],
        "ccdVisitId": numpy.full(nrows, ccdVisitId, dtype=numpy.int64),
        "midPointTai": numpy.full(nrows, midPointTai, dtype=numpy.float64),
        "flags": numpy.full(nrows, 0, dtype=numpy.int64),
    })
    return df

makeObjectCatalog()

pandas.DataFrame lsst.dax.apdb.tests.data_factory.makeObjectCatalog	(	Region	region,
		int	count
	)

Make a catalog containing a bunch of DiaObjects inside a region.

Parameters
----------
region : `lsst.sphgeom.Region`
    Spherical region.
count : `int`
    Number of records to generate.

Returns
-------
catalog : `pandas.DataFrame`
    Catalog of DiaObjects records.

Notes
-----
Returned catalog only contains three columns - ``diaObjectId`, ``ra``, and
``decl`` (in degrees).

Definition at line 64 of file data_factory.py.

def makeObjectCatalog(region: Region, count: int) -> pandas.DataFrame:
    """Make a catalog containing a bunch of DiaObjects inside a region.
 
    Parameters
    ----------
    region : `lsst.sphgeom.Region`
        Spherical region.
    count : `int`
        Number of records to generate.
 
    Returns
    -------
    catalog : `pandas.DataFrame`
        Catalog of DiaObjects records.
 
    Notes
    -----
    Returned catalog only contains three columns - ``diaObjectId`, ``ra``, and
    ``decl`` (in degrees).
    """
    points = list(_genPointsInRegion(region, count))
    # diaObjectId=0 may be used in some code for DiaSource foreign key to mean
    # the same as ``None``.
    ids = numpy.arange(1, len(points) + 1, dtype=numpy.int64)
    ras = numpy.array([sp.getRa().asDegrees() for sp in points], dtype=numpy.float64)
    decls = numpy.array([sp.getDec().asDegrees() for sp in points], dtype=numpy.float64)
    df = pandas.DataFrame({"diaObjectId": ids,
                           "ra": ras,
                           "decl": decls})
    return df
 
 

makeSourceCatalog()

pandas.DataFrame lsst.dax.apdb.tests.data_factory.makeSourceCatalog	(	pandas.DataFrame	objects,
		DateTime	visit_time,
		int	start_id = 0,
		int	ccdVisitId = 1
	)

Make a catalog containing a bunch of DiaSources associated with the
input DiaObjects.

Parameters
----------
objects : `pandas.DataFrame`
    Catalog of DiaObject records.
visit_time : `lsst.daf.base.DateTime`
    Time of the visit.
start_id : `int`
    Starting value for ``diaObjectId``.
ccdVisitId : `int`
    Value for ``ccdVisitId`` field.

Returns
-------
catalog : `pandas.DataFrame`
    Catalog of DiaSource records.

Notes
-----
Returned catalog only contains small number of columns needed for tests.

Definition at line 96 of file data_factory.py.

                      start_id: int = 0, ccdVisitId: int = 1) -> pandas.DataFrame:
    """Make a catalog containing a bunch of DiaSources associated with the
    input DiaObjects.
 
    Parameters
    ----------
    objects : `pandas.DataFrame`
        Catalog of DiaObject records.
    visit_time : `lsst.daf.base.DateTime`
        Time of the visit.
    start_id : `int`
        Starting value for ``diaObjectId``.
    ccdVisitId : `int`
        Value for ``ccdVisitId`` field.
 
    Returns
    -------
    catalog : `pandas.DataFrame`
        Catalog of DiaSource records.
 
    Notes
    -----
    Returned catalog only contains small number of columns needed for tests.
    """
    nrows = len(objects)
    midPointTai = visit_time.get(system=DateTime.MJD)
    df = pandas.DataFrame({
        "diaSourceId": numpy.arange(start_id, start_id + nrows, dtype=numpy.int64),
        "diaObjectId": objects["diaObjectId"],
        "ccdVisitId": numpy.full(nrows, ccdVisitId, dtype=numpy.int64),
        "parentDiaSourceId": 0,
        "ra": objects["ra"],
        "decl": objects["decl"],
        "midPointTai": numpy.full(nrows, midPointTai, dtype=numpy.float64),
        "flags": numpy.full(nrows, 0, dtype=numpy.int64),
    })
    return df