'''test06_get.py
Test Data class get methods
CS 251/2: Data Analysis and Visualization
Fall 2026
Oliver Layton, Caitrin Eaton, Hannah Wolfe, Stephanie Taylor
'''
import numpy as np

from data import Data


def test_all_get_methods(iris_filename):
    iris_data = Data(iris_filename)

    print(f"Your iris headers are\n{iris_data.get_headers()}\nand should be\n['sepal_length', 'sepal_width', 'petal_length', 'petal_width', 'species']\n")
    print(f"Your iris variable mapping is\n{iris_data.get_mappings()}\nand should be\n{{'sepal_length': 0, 'sepal_width': 1, 'petal_length': 2, 'petal_width': 3, 'species': 4}}\n")
    print(f'Your data has {iris_data.get_num_samples()} samples and {iris_data.get_num_dims()} variables/dimensions.\nIt should have 150 samples and 5 variables/dimensions.\n')
    print(f'Your 10th sample is\n{iris_data.get_sample(9)}\nand should be \n[4.9 3.1 1.5 0.1 0. ]\n')
    print(f'The indices of the headers sepal_width and petal_width are\n{iris_data.get_header_indices(["sepal_width", "petal_width"])}\nand should be \n[1, 3]\n')


if __name__ == '__main__':
    print('---------------------------------------------------------------------------------------')
    print('Beginning test 1 (Test all get methods)...')
    print('---------------------------------------------')
    data_file = 'data/iris.csv'
    test_all_get_methods(data_file)
    print('---------------------------------------------')
    print('Finished test 1!')
    print('---------------------------------------------------------------------------------------')