Add more unit tests #5

	name: Run tests on merge

	on:
	pull_request:
	branches:
	- main

	jobs:
	test:
	runs-on: ubuntu-latest

	steps:
	# Checkout the repository code
	- name: Checkout code
	uses: actions/checkout@v3

	# Set up Python environment
	- name: Set up Python
	uses: actions/setup-python@v4
	with:
	python-version: '3.10' # Use the Python version your project needs

	# Install dependencies
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -r requirements.txt
	pip install lightgbm xgboost catboost # Install the libraries required for tests
	pip install pytest

	# Run tests using pytest
	- name: Run tests
	run: \|
	pytest --maxfail=1 --disable-warnings

Provide feedback