pyunit_colname_set_basic.py

from __future__ import print_function
import sys
sys.path.insert(1,"../../")
import h2o
from tests import pyunit_utils




def colname_set_basic():
    
    

    print("Uploading iris data...")

    no_headers = h2o.upload_file(pyunit_utils.locate("smalldata/iris/iris.csv"))
    headers_and = h2o.upload_file(pyunit_utils.locate("smalldata/iris/iris_header.csv"))

    print(no_headers.names)
    print(headers_and.names)

    no_headers.set_names(headers_and.names)
    assert no_headers.names == headers_and.names, "Expected the same column names but got {0} and {1}".\
        format(no_headers.names, headers_and.names)



if __name__ == "__main__":
    pyunit_utils.standalone_test(colname_set_basic)
else:
    colname_set_basic()