def get_classes(filename, class_name): if class_name: return [ i for i in _collect_classes(filename) if class_name == get_class_name(i) ] else: return _collect_classes(filename)
def get_classes(filename, class_name): if class_name: classes = [i for i in _collect_classes(filename) if class_name == get_class_name(i)] else: classes = [i for i in _collect_classes(filename)] return classes
name = "Case_foo_bar" assert util.name_convertion(name) == 'foo bar' it "does not remove case if not uppercase": name = "case_foo_bar" assert util.name_convertion(name) == 'Case foo bar' describe "get class name": it "returns itself when it str does not start with case": name = "a string" assert util.get_class_name(name) == name it "returns none when does not start with Case": name = """<class '_stdout_messages_for_the_command_line'>""" assert util.get_class_name(name) == None it "parses correctly when a valid class name is sent": name = """<class 'Case_stdout_messages_for_the_command_line'>""" assert util.get_class_name(name) == 'Case_stdout_messages_for_the_command_line' #describe "stop watch": #