gourav/PyFacts
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Math tests and series dtype param name change

Browse Source
This commit is contained in:
Gourav Kumar 2022-04-12 11:43:52 +05:30
parent 49604a5ae9
commit e8be7e9efa
1 changed files with 30 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tests/test_core.py
View File

@ -86,12 +86,12 @@ class TestAllFrequencies:
class TestSeries: class TestSeries:
def test_creation(self): def test_creation(self):
series = Series([1, 2, 3, 4, 5, 6, 7], data_type="number") series = Series([1, 2, 3, 4, 5, 6, 7], dtype="number")
assert series.dtype == float assert series.dtype == float
assert series[2] == 3 assert series[2] == 3
dates = create_date_series("2021-01-01", "2021-01-31", frequency="D") dates = create_date_series("2021-01-01", "2021-01-31", frequency="D")
series = Series(dates, data_type="date") series = Series(dates, dtype="date")
assert series.dtype == datetime.datetime assert series.dtype == datetime.datetime
@ -292,7 +292,7 @@ class TestTimeSeriesComparisons:
def test_series_comparison(self): def test_series_comparison(self):
ts1 = TimeSeriesCore(self.data1, "M") ts1 = TimeSeriesCore(self.data1, "M")
ser = Series([240, 210, 240, 270], data_type="int") ser = Series([240, 210, 240, 270], dtype="int")
assert (ts1 > ser).values == Series([0.0, 1.0, 0.0, 0.0], "float") assert (ts1 > ser).values == Series([0.0, 1.0, 0.0, 0.0], "float")
assert (ts1 >= ser).values == Series([0.0, 1.0, 1.0, 0.0], "float") assert (ts1 >= ser).values == Series([0.0, 1.0, 1.0, 0.0], "float")
@ -315,8 +315,8 @@ class TestTimeSeriesComparisons:
def test_errors(self): def test_errors(self):
ts1 = TimeSeriesCore(self.data1, "M") ts1 = TimeSeriesCore(self.data1, "M")
ts2 = TimeSeriesCore(self.data2, "M") ts2 = TimeSeriesCore(self.data2, "M")
ser = Series([240, 210, 240], data_type="int") ser = Series([240, 210, 240], dtype="int")
ser2 = Series(["2021-01-01", "2021-02-01", "2021-03-01", "2021-04-01"], data_type="date") ser2 = Series(["2021-01-01", "2021-02-01", "2021-03-01", "2021-04-01"], dtype="date")
del ts2["2021-04-01"] del ts2["2021-04-01"]
@ -334,3 +334,28 @@ class TestTimeSeriesComparisons:
with pytest.raises(TypeError): with pytest.raises(TypeError):
ts2 < [23, 24, 25, 26] ts2 < [23, 24, 25, 26]
class TestTimeSeriesArithmatic:
data = [
("2021-01-01", 220),
("2021-02-01", 230),
("2021-03-01", 240),
("2021-04-01", 250),
]
def test_add(self):
ts = TimeSeriesCore(self.data, "M")
ser = ts.values
num_add_ts = ts + 40
assert num_add_ts["2021-01-01"][1] == 260
assert num_add_ts["2021-04-01"][1] == 290
num_radd_ts = 40 + ts
assert num_radd_ts["2021-01-01"][1] == 260
assert num_radd_ts["2021-04-01"][1] == 290
ser_add_ts = ts + ser
assert ser_add_ts["2021-01-01"][1] == 440
assert ser_add_ts["2021-04-01"][1] == 500