Python py_get_unit_from_dtype 예제들

예제 #1

def test_get_unit_from_dtype():
    # datetime64
    assert py_get_unit_from_dtype(
        np.dtype("M8[Y]")) == NpyDatetimeUnit.NPY_FR_Y.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[M]")) == NpyDatetimeUnit.NPY_FR_M.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[W]")) == NpyDatetimeUnit.NPY_FR_W.value
    # B has been deprecated and removed -> no 3
    assert py_get_unit_from_dtype(
        np.dtype("M8[D]")) == NpyDatetimeUnit.NPY_FR_D.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[h]")) == NpyDatetimeUnit.NPY_FR_h.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[m]")) == NpyDatetimeUnit.NPY_FR_m.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[s]")) == NpyDatetimeUnit.NPY_FR_s.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[ms]")) == NpyDatetimeUnit.NPY_FR_ms.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[us]")) == NpyDatetimeUnit.NPY_FR_us.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[ns]")) == NpyDatetimeUnit.NPY_FR_ns.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[ps]")) == NpyDatetimeUnit.NPY_FR_ps.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[fs]")) == NpyDatetimeUnit.NPY_FR_fs.value
    assert py_get_unit_from_dtype(
        np.dtype("M8[as]")) == NpyDatetimeUnit.NPY_FR_as.value

    # timedelta64
    assert py_get_unit_from_dtype(
        np.dtype("m8[Y]")) == NpyDatetimeUnit.NPY_FR_Y.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[M]")) == NpyDatetimeUnit.NPY_FR_M.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[W]")) == NpyDatetimeUnit.NPY_FR_W.value
    # B has been deprecated and removed -> no 3
    assert py_get_unit_from_dtype(
        np.dtype("m8[D]")) == NpyDatetimeUnit.NPY_FR_D.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[h]")) == NpyDatetimeUnit.NPY_FR_h.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[m]")) == NpyDatetimeUnit.NPY_FR_m.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[s]")) == NpyDatetimeUnit.NPY_FR_s.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[ms]")) == NpyDatetimeUnit.NPY_FR_ms.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[us]")) == NpyDatetimeUnit.NPY_FR_us.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[ns]")) == NpyDatetimeUnit.NPY_FR_ns.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[ps]")) == NpyDatetimeUnit.NPY_FR_ps.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[fs]")) == NpyDatetimeUnit.NPY_FR_fs.value
    assert py_get_unit_from_dtype(
        np.dtype("m8[as]")) == NpyDatetimeUnit.NPY_FR_as.value

예제 #2

파일: test_np_datetime.py 프로젝트: selasley/pandas

def test_get_unit_from_dtype():
    # datetime64
    assert py_get_unit_from_dtype(np.dtype("M8[Y]")) == 0
    assert py_get_unit_from_dtype(np.dtype("M8[M]")) == 1
    assert py_get_unit_from_dtype(np.dtype("M8[W]")) == 2
    # B has been deprecated and removed -> no 3
    assert py_get_unit_from_dtype(np.dtype("M8[D]")) == 4
    assert py_get_unit_from_dtype(np.dtype("M8[h]")) == 5
    assert py_get_unit_from_dtype(np.dtype("M8[m]")) == 6
    assert py_get_unit_from_dtype(np.dtype("M8[s]")) == 7
    assert py_get_unit_from_dtype(np.dtype("M8[ms]")) == 8
    assert py_get_unit_from_dtype(np.dtype("M8[us]")) == 9
    assert py_get_unit_from_dtype(np.dtype("M8[ns]")) == 10
    assert py_get_unit_from_dtype(np.dtype("M8[ps]")) == 11
    assert py_get_unit_from_dtype(np.dtype("M8[fs]")) == 12
    assert py_get_unit_from_dtype(np.dtype("M8[as]")) == 13

    # timedelta64
    assert py_get_unit_from_dtype(np.dtype("m8[Y]")) == 0
    assert py_get_unit_from_dtype(np.dtype("m8[M]")) == 1
    assert py_get_unit_from_dtype(np.dtype("m8[W]")) == 2
    # B has been deprecated and removed -> no 3
    assert py_get_unit_from_dtype(np.dtype("m8[D]")) == 4
    assert py_get_unit_from_dtype(np.dtype("m8[h]")) == 5
    assert py_get_unit_from_dtype(np.dtype("m8[m]")) == 6
    assert py_get_unit_from_dtype(np.dtype("m8[s]")) == 7
    assert py_get_unit_from_dtype(np.dtype("m8[ms]")) == 8
    assert py_get_unit_from_dtype(np.dtype("m8[us]")) == 9
    assert py_get_unit_from_dtype(np.dtype("m8[ns]")) == 10
    assert py_get_unit_from_dtype(np.dtype("m8[ps]")) == 11
    assert py_get_unit_from_dtype(np.dtype("m8[fs]")) == 12
    assert py_get_unit_from_dtype(np.dtype("m8[as]")) == 13

예제 #3

파일: timedeltas.py 프로젝트: selasley/pandas

 def _reso(self):
     return py_get_unit_from_dtype(self.dtype)