fix some imports

This commit is contained in:
Dr. Kashif Rasul
2019-11-05 16:43:18 +01:00
parent 7d5e808a6c
commit 8415a522ec
8 changed files with 339 additions and 7 deletions
+20
View File
@@ -0,0 +1,20 @@
import pytest
import pandas as pd
from pts.dataset import ProcessStartField
@pytest.mark.parametrize(
"freq, expected",
[
("B", "2019-11-01"),
("W", "2019-11-03"),
("M", "2019-11-30"),
("12M", "2019-11-30"),
("A-DEC", "2019-12-31"),
],
)
def test_process_start_field(freq, expected):
process = ProcessStartField.process
given = "2019-11-01 12:34:56"
assert process(given, freq) == pd.Timestamp(expected, freq)