added other split_type options i.e none and none with bts

scripts/ml/ml_functions/test_func_singlegene.py

@@ -109,3 +109,24 @@ baz_df1.rename(columns = {0:'vdw_scaled', 1: 'ele_scaled'}, inplace = True)
 baz2 = pd.concat([baz, baz_df1], axis = 1)
 
 a = pd.concat([bar2, baz2], axis = 1)
+
+#%% test added split_types i.e none_with_bts and none_only
+
+spl_type = 'none_with_bts'
+spl_type = 'none_only'
+
+#data_type  = "actual"
+data_type = "complete"
+
+df2 = split_tts(df
+          , data_type = data_type # only works with complete despite what you set to
+          , split_type = spl_type
+          , oversampling = True
+          , dst_colname = 'dst'
+          , target_colname = 'dst_mode'
+          , include_gene_name = True
+          , random_state = 42 # default
+      )
+
+all(df2['X'].columns.isin(['gene_name'])) # should be False
+