mirror of
https://github.com/elastic/eland.git
synced 2025-07-24 00:00:39 +08:00
28 lines
799 B
Python
28 lines
799 B
Python
# File called _pytest for PyCharm compatability
|
|
import eland as ed
|
|
import pandas as pd
|
|
from eland.tests import ELASTICSEARCH_HOST
|
|
from eland.tests import FLIGHTS_INDEX_NAME, ECOMMERCE_INDEX_NAME
|
|
from eland.tests.common import TestData
|
|
|
|
|
|
class TestSeriesRepr(TestData):
|
|
|
|
def test_repr_flights_carrier(self):
|
|
pd_s = self.pd_flights()['Carrier']
|
|
ed_s = ed.Series(ELASTICSEARCH_HOST, FLIGHTS_INDEX_NAME, 'Carrier')
|
|
|
|
pd_repr = repr(pd_s)
|
|
ed_repr = repr(ed_s)
|
|
|
|
assert pd_repr == ed_repr
|
|
|
|
def test_repr_flights_carrier_5(self):
|
|
pd_s = self.pd_flights()['Carrier'].head(5)
|
|
ed_s = ed.Series(ELASTICSEARCH_HOST, FLIGHTS_INDEX_NAME, 'Carrier').head(5)
|
|
|
|
pd_repr = repr(pd_s)
|
|
ed_repr = repr(ed_s)
|
|
|
|
assert pd_repr == ed_repr
|