update to tests.

solventmapcreator/test/solvationcalculationtest/solvationplotinformationtest.py

@@ -38,12 +38,14 @@ class SolvationPlotInformationTestCase(unittest.TestCase):
                                             [-8.004792, -7.545754]]),
                          "x_label":"x_label", "y_label":"y_label",
                          "plot_axis_range":(2.0, 0.0, 0.0, 1.0),
-                         "levels":[2*x -31.0 for x in range(32)]}
+                         "levels":[2*x -31 for x in range(32)]}
         actual_dict = solvationplotinformation.create_plot_input_data(self.epsilon_i_list, self.epsilon_j_list, 298.0, 1.0, np.array([0.5, 1.0]), "x_label", "y_label")
         self.assertListEqual(sorted(expected_dict.keys()), sorted(actual_dict.keys()))
         for key in expected_dict.keys():
-            if key == "plot_axis_range" or "levels" == key:
+            if key == "plot_axis_range":
                 np.testing.assert_array_almost_equal(np.array(expected_dict[key]), np.array(actual_dict[key]))
+            elif  "levels" == key:
+                np.testing.assert_array_equal(np.array(expected_dict[key]), np.array(actual_dict[key]))
             elif "_data" in key:
                 np.testing.assert_array_almost_equal(expected_dict[key], actual_dict[key])
             else:
@@ -53,12 +55,14 @@ class SolvationPlotInformationTestCase(unittest.TestCase):
         """
         expected_dict = {"x_label":"x_label", "y_label":"y_label",
                          "plot_axis_range":(2.0, 0.0, 0.0, 1.0),
-                         "levels":[2*x -31.0 for x in range(32)]}
+                         "levels":[2*x -31 for x in range(32)]}
         actual_dict = solvationplotinformation.create_input_data_labels(self.epsilon_i_list, self.epsilon_j_list, "x_label", "y_label")
         self.assertListEqual(sorted(expected_dict.keys()), sorted(actual_dict.keys()))
         for key in expected_dict.keys():
-            if key == "plot_axis_range" or key == "levels":
+            if key == "plot_axis_range":
                 np.testing.assert_array_almost_equal(np.array(expected_dict[key]), np.array(actual_dict[key]))
+            elif  "levels" == key:
+                np.testing.assert_array_equal(np.array(expected_dict[key]), np.array(actual_dict[key]))
             else:
                 self.assertEqual(expected_dict[key], actual_dict[key])
     def test_create_input_data_values(self):