Add test to test expand_lumis
Account for key=lumi in count function