def test_individual_strings(self):
base_folder = ""
test_input = [
# simplest case
{"input": [
"azide_intensity_175815_count_ds0_sp0_sm0_de0_du0_0.csv",
],
"result": 1,
"base_filename":"scan",
"extension":".csv",
"test": "equal"},
# fail test
{"input": [
"azide_intensity_175815_count_ds0_sp0_sm0_de0_du0_0.csv",
],
"result": 2,
"base_filename":"scan",
"extension":".csv",
"test": "not equal"},
# simple case
{"input": [
"azide_intensity_175815_count_ds0_sp0_sm0_de0_du0_0.csv",
"azide_intensity_175815_count_ds0_sp0_sm0_de0_du0_1.csv",
"azide_intensity_175815_count_ds1_sp0_sm0_de0_du0_0.csv",
"azide_intensity_175815_count_ds1_sp0_sm0_de0_du0_1.csv",
"azide_intensity_175815_intensity_ds0_sp0_sm0_de0_du0_0.csv",
"azide_intensity_175815_intensity_ds0_sp0_sm0_de0_du0_1.csv",
"azide_intensity_175815_intensity_ds1_sp0_sm0_de0_du0_0.csv",
"azide_intensity_175815_intensity_ds1_sp0_sm0_de0_du0_1.csv",
],
"result": 2,
"base_filename":"scan",
"extension":".csv",
"test": "equal"},
# long filenames
{"input": [
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_count_ds0_sp0_sm0_de0_du0_0.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_count_ds0_sp0_sm0_de0_du0_1.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_count_ds1_sp0_sm0_de0_du0_0.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_count_ds1_sp0_sm0_de0_du0_1.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_intensity_ds0_sp0_sm0_de0_du0_0.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_intensity_ds0_sp0_sm0_de0_du0_1.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_intensity_ds1_sp0_sm0_de0_du0_0.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_intensity_ds1_sp0_sm0_de0_du0_1.csv",
],
"result": 2,
"base_filename":"scan",
"extension":".csv",
"test": "equal"},
# long filenames, higher numbers
{"input": [
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_count_ds10_sp0_sm0_de0_du0_0.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_count_ds10_sp0_sm0_de0_du0_1.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_count_ds11_sp0_sm0_de0_du0_0.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_count_ds11_sp0_sm0_de0_du0_1.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_intensity_ds10_sp0_sm0_de0_du0_0.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_intensity_ds10_sp0_sm0_de0_du0_1.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_intensity_ds11_sp0_sm0_de0_du0_0.csv",
"/Users/robbert/Desktop/20151006/azide_intensity_175815/azide_intensity_175815_intensity_ds11_sp0_sm0_de0_du0_1.csv",
],
"result": 12,
"base_filename":"scan",
"extension":".csv",
"test": "equal"},
]
for i in range(len(test_input)):
res = IOM.find_number_of_datastates(base_folder = "", flag_verbose = self.flag_verbose, test_input = test_input[i]["input"])
if test_input[i]["test"] == "equal":
self.assertEqual(res, test_input[i]["result"])
elif test_input[i]["test"] == "not equal":
self.assertNotEqual(res, test_input[i]["result"])
else:
print("Invalid test")
