1
0
Fork 0

Update test_sample_importer.py

master
lmoretti96 4 years ago committed by GitHub
parent 1f7c0d5e59
commit 8ed694fb28
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 5
      PyCTBN/tests/utility/test_sample_importer.py

@ -17,9 +17,10 @@ class TestSampleImporter(unittest.TestCase):
def setUpClass(cls) -> None:
with open("./PyCTBN/test_data/networks_and_trajectories_binary_data_01_3.json") as f:
raw_data = json.load(f)
#read the samples
trajectory_list_raw= raw_data[0]["samples"]
cls.trajectory_list = [pd.DataFrame(sample) for sample in trajectory_list_raw]
cls.variables= pd.DataFrame(raw_data[0]["variables"])