import sys

import os

import pandas as pd

# Add the '/scripts' directory to the Python path

sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../scripts')))

from data.preparation import feature_selection

from data.preparation import data_split

# Test feature selection function

def test_feature_selection():

    for features in feature_selection().items():

        assert features

# Test data split function

def test_data_split():

    os.makedirs('../../data/input', exist_ok=True)

    assert data_split(feature_selection().drop('LUNG_CANCER', axis='columns'), feature_selection()['LUNG_CANCER'])