def test_bert_embeddings(): event = nuclio.Event(body=json.dumps(['John loves Mary'])) ctx = nuclio.Context() init_context(ctx) outputs = pickle.loads(handler(ctx, event)) assert (True if abs(np.mean(outputs[0]) - -0.011996539) <= 0.0001 else False) is True assert (True if abs(np.mean(outputs[0]) - -0.011996539) > 0 else False) is True
def fake_nuclio_context(body, headers=None): return nuclio.Context(), nuclio.Event(body=body, headers=headers)
def test_bert_embeddings(): event = nuclio.Event(body=json.dumps(['John loves Mary'])) ctx = nuclio.Context() init_context(ctx) outputs = pickle.loads(handler(ctx, event)) print(outputs)