@@ -28,22 +28,19 @@ def writer_test_city( writer : Callable
28
28
with tb .open_file (file_out , 'w' ) as h5out :
29
29
writer (h5out )
30
30
31
- def get_writers (self , h5out ): pass
32
- def write_parameters (self , h5out ): pass
33
-
34
31
35
32
def _df_writer (h5out ):
36
33
df = pd .DataFrame (columns = ['event' , 'some_value' ], dtype = int )
37
34
return df_writer (h5out , df , 'DUMMY' , 'dummy' , columns_to_index = ['event' ])
38
35
39
36
40
- @mark .parametrize (" writer group node column thing" .split (),
41
- [( hits_writer , "RECO" , "Events" , "event" , "hits" ),
42
- ( kr_writer , "DST" , "Events" , "event" , "kr" ),
43
- ( pmap_writer , "PMAPS" , "S1" , "event" , "s1" ),
44
- ( pmap_writer , "PMAPS" , "S2" , "event" , "s2" ),
45
- ( pmap_writer , "PMAPS" , "S2Si" , "event" , "s2si" ),
46
- ( _df_writer , "DUMMY" , "dummy" , "event" , 'df' )])
37
+ @mark .parametrize (" writer group node column thing" .split (),
38
+ [( hits_writer , "RECO" , "Events" , "event" , "hits" ),
39
+ ( kr_writer , "DST" , "Events" , "event" , "kr" ),
40
+ ( pmap_writer , "PMAPS" , "S1" , "event" , "s1" ),
41
+ ( pmap_writer , "PMAPS" , "S2" , "event" , "s2" ),
42
+ ( pmap_writer , "PMAPS" , "S2Si" , "event" , "s2si" ),
43
+ ( _df_writer , "DUMMY" , "dummy" , "event" , 'df' )])
47
44
def test_table_is_indexed (tmpdir_factory , writer , group , node , column , thing ):
48
45
tmpdir = tmpdir_factory .mktemp ('indexation' )
49
46
file_out = os .path .join (tmpdir , f"empty_table_containing_{ thing } .h5" )
0 commit comments