mirror of
https://github.com/elastic/eland.git
synced 2025-07-11 00:02:14 +08:00
45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
# File called _pytest for PyCharm compatability
|
|
|
|
from pandas.util.testing import assert_series_equal
|
|
|
|
from eland.tests.common import TestData
|
|
|
|
|
|
class TestDataFrameMetrics(TestData):
|
|
|
|
def test_mean(self):
|
|
pd_flights = self.pd_flights()
|
|
ed_flights = self.ed_flights()
|
|
|
|
pd_mean = pd_flights.mean(numeric_only=True)
|
|
ed_mean = ed_flights.mean(numeric_only=True)
|
|
|
|
assert_series_equal(pd_mean, ed_mean)
|
|
|
|
def test_sum(self):
|
|
pd_flights = self.pd_flights()
|
|
ed_flights = self.ed_flights()
|
|
|
|
pd_sum = pd_flights.sum(numeric_only=True)
|
|
ed_sum = ed_flights.sum(numeric_only=True)
|
|
|
|
assert_series_equal(pd_sum, ed_sum)
|
|
|
|
def test_min(self):
|
|
pd_flights = self.pd_flights()
|
|
ed_flights = self.ed_flights()
|
|
|
|
pd_min = pd_flights.min(numeric_only=True)
|
|
ed_min = ed_flights.min(numeric_only=True)
|
|
|
|
assert_series_equal(pd_min, ed_min)
|
|
|
|
def test_max(self):
|
|
pd_flights = self.pd_flights()
|
|
ed_flights = self.ed_flights()
|
|
|
|
pd_max = pd_flights.max(numeric_only=True)
|
|
ed_max = ed_flights.max(numeric_only=True)
|
|
|
|
assert_series_equal(pd_max, ed_max)
|