def METHOD_NAME():
    """Test the functionality of the ElasticsearchIODataset"""
    dataset = tfio.experimental.elasticsearch.ElasticsearchIODataset(
        nodes=[NODE], index=INDEX, doc_type=DOC_TYPE, headers=HEADERS
    )
    assert issubclass(type(dataset), tf.data.Dataset)
    for item in dataset:
        for attr in ATTRS:
            assert attr in item