Digital Drosophila Archive
30 lines
1.5 KiB
Python
30 lines
1.5 KiB
Python
import tempfile
|
|
from pathlib import Path
|
|
import unittest
|
|
import numpy as np
|
|
import pandas as pd
|
|
import pyarrow as pa
|
|
import pyarrow.feather as feather
|
|
from audit import raw_incident,population,save
|
|
|
|
class AuditTests(unittest.TestCase):
|
|
def test_raw_incident_union_keeps_self_loop_once(self):
|
|
with tempfile.TemporaryDirectory(prefix='apl-scope-test-') as d:
|
|
p=Path(d)/'raw.feather'
|
|
frame=pd.DataFrame({'body_pre':[10,10,20,40],'body_post':[10,30,10,50],'weight':[1,2,3,4]},dtype=np.int64)
|
|
feather.write_feather(pa.Table.from_pandas(frame,preserve_index=False),p,chunksize=2)
|
|
result=raw_incident(p,np.array([10],np.uint64))
|
|
self.assertTrue(np.array_equal(result.to_numpy(),frame.iloc[:3].to_numpy()))
|
|
with self.assertRaises(AssertionError):
|
|
feather.write_feather(pa.table({'body_pre':[10],'wrong':[30],'weight':[1]}),p)
|
|
raw_incident(p,np.array([10],np.uint64))
|
|
def test_population_does_not_equate_dopamine_to_DAN(self):
|
|
m=pd.DataFrame({'annotation_class':[None,None,'ALPN','DAN'], 'annotation_type':['APL','DPM','VA6_adPN','PAM01']})
|
|
self.assertEqual(list(population(m)),['APL','DPM','ALPN','DAN'])
|
|
def test_exclusive_output(self):
|
|
with tempfile.TemporaryDirectory(prefix='apl-scope-test-') as d:
|
|
p=Path(d)/'x.json';save(p,{'original':True})
|
|
with self.assertRaises(FileExistsError):save(p,{'original':False})
|
|
|
|
if __name__=='__main__':unittest.main()
|