mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-07-11 04:26:09 +08:00
Dr. Kashif Rasul
17 lines
551 B
Python
17 lines
551 B
Python
from pts.dataset import FieldName
|
|
|
|
|
|
def test_dataset_fields():
|
|
assert (
|
|
"feat_static_cat" == FieldName.FEAT_STATIC_CAT
|
|
), "Error in the FieldName 'feat_static_cat'."
|
|
assert (
|
|
"feat_static_real" == FieldName.FEAT_STATIC_REAL
|
|
), "Error in the FieldName 'feat_static_real'."
|
|
assert (
|
|
"feat_dynamic_cat" == FieldName.FEAT_DYNAMIC_CAT
|
|
), "Error in the FieldName 'feat_dynamic_cat'."
|
|
assert (
|
|
"feat_dynamic_real" == FieldName.FEAT_DYNAMIC_REAL
|
|
), "Error in the FieldName 'feat_dynamic_real'."
|