maaikeb / elasticsearch-benchmarking Goto Github PK

class Result(dict):

    def position(self):
        raise NotImplementedError()


class ElasticSearchResult(Result):
    pass


class Query(object):
    pass


class ElasticSearchQuery(Query):

    def __init__(self, term, filters={})
        pass


class BaseResultOracle(object):
    def query(self, query: Query) -> Result[]:
        raise NotImplementedError()


class ElasticSearchOracle(BaseResultOracle):

    def query(self, query: ElasticSearchQuery):
        # returns ElasticSearchResult[]

class BaseScorer(object):
    def calculate_score(self, matcher_results) -> float:
        raise NotImplementedError()

class FirstPageScorer(BaseScorer):
    pass
    
    
b = Benchmark(
    oracle=ElasticSearchOracle(hostname, port),
    scorer=FirstPageScorer(page_size=10, factor=0.87),
)

test_b = Benchmark(
    oracle=FixedResultsOracle(test_results),
    scorer=ConstScorer(5),
)

test_suite = [
    (ElasticSearchQuery('JK Rowling'), lambda r: r['author'] == 'JK Rowling'),
    (ElasticSearchQuery('Harry Potter 1'), lambda r: r['isbn'] == '12331234657465'),
]

score = b.execute(test_suite)

שיקום
returns results with הקמה instead of שיקום

קוד פתוח
returns results with only פתוח