53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
import os
|
|
import sys
|
|
import unittest
|
|
import json
|
|
import numpy as np
|
|
import pandas as pd
|
|
|
|
PROJECT_ROOT = os.path.realpath(os.path.join(__file__, "..", "..", ".."))
|
|
MODULE_ROOT = os.path.join(PROJECT_ROOT, "modules")
|
|
sys.path.append(MODULE_ROOT)
|
|
print(MODULE_ROOT)
|
|
|
|
from pandas_wrapper.utils import index_to_jsons, jsons_to_index
|
|
|
|
class TestIndexJsonConversion(unittest.TestCase):
|
|
def setUp(self):
|
|
"""Set up test data"""
|
|
self.index_int = pd.Index([1, 2, 3], name="test_index_int")
|
|
self.index_float = pd.Index([1.1, 2.2, 3.3], name="test_index_float")
|
|
self.index_str = pd.Index(["a", "b", "c"], name="test_index_str")
|
|
self.index_mixed = pd.Index([1, "text", 3.5], name="test_index_mixed")
|
|
self.index_no_name = pd.Index([1, "text", 3.5])
|
|
|
|
def test_index_to_jsons_and_back(self):
|
|
"""Test conversion from Index to JSON and back"""
|
|
for index in [self.index_int, self.index_float, self.index_str, self.index_mixed]:
|
|
with self.subTest(index=index):
|
|
json_str = index_to_jsons(index)
|
|
reconstructed_index = jsons_to_index(json_str)
|
|
self.assertTrue(index.equals(reconstructed_index), f"Failed for index: {index.name}")
|
|
|
|
def test_json_format(self):
|
|
"""Test if JSON output is correctly formatted"""
|
|
json_str = index_to_jsons(self.index_int)
|
|
data = json.loads(json_str)
|
|
|
|
self.assertIn("index", data)
|
|
self.assertIn("name", data)
|
|
self.assertEqual(data["name"], "test_index_int")
|
|
|
|
def test_json_format_no_name(self):
|
|
"""Test if JSON output is correctly formatted"""
|
|
json_str = index_to_jsons(self.index_no_name)
|
|
data = json.loads(json_str)
|
|
|
|
self.assertIn("index", data)
|
|
self.assertIn("name", data)
|
|
self.assertTrue(data["name"] is None)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|