test for data_load module

tests/data_prep/test_conversion.py

@@ -0,0 +1,25 @@
+from data_prep.data_load import prepare_data , dataframe_to_tensor
+
+import pandas as pd
+import torch
+
+test_file = "data/NIFTY_5_years.csv"
+
+
+def test_prepare_data():
+
+  df = prepare_data(test_file)
+  
+  assert type(df) == pd.DataFrame
+
+
+def test_convert_to_tensor():
+
+  df = prepare_data(test_file)
+
+  tensorA = dataframe_to_tensor(df)
+
+  assert type(tensorA) == torch.Tensor
+
+
+