🐛 fix remaining colors, test

vaep/plotting/defaults.py

@@ -31,7 +31,7 @@
 color_model_mapping['MSIMPUTE_MNAR'] = other_colors[7]
 color_model_mapping['TRKNN'] = other_colors[8]
 color_model_mapping['SVDMETHOD'] = other_colors[9]
-other_colors = sns.color_palette()[10:]
+other_colors = other_colors[10:]
 
 
 def assign_colors(models):

vaep/tests/plotting/test_defaults.py

@@ -0,0 +1,9 @@
+from vaep.plotting.defaults import assign_colors
+
+
+def test_assign_colors():
+    expected = [(0.8392156862745098, 0.15294117647058825, 0.1568627450980392),
+                (0.17254901960784313, 0.6274509803921569, 0.17254901960784313),
+                (0.21044753832183283, 0.6773105080456748, 0.6433941168468681)]
+    assigned = assign_colors(['DAE', 'CF', 'Test'])
+    assert assigned == expected