21 lines
518 B
Python
21 lines
518 B
Python
import unittest
|
|
|
|
import numpy as np
|
|
import pandas as pd
|
|
from numpy.random import default_rng
|
|
from pandas.testing import assert_series_equal
|
|
|
|
from features.esm import preprocess_esm
|
|
|
|
|
|
class EsmFeatures(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
cls.esm = pd.DataFrame(
|
|
{"double_esm_user_answer_timestamp": [1622127860000, 1622129860000]}
|
|
)
|
|
|
|
def test_preprocess_esm(self):
|
|
self.esm_processed = preprocess_esm(self.esm)
|
|
print(self.esm_processed)
|