def test_as_sorted_flat_values(self): """tests that the flat values of the input matrices are all put in one big numpy array""" m1 = dm.DataMatrix(2, 2, values=[[2, np.nan], [3, 4]]) m2 = dm.DataMatrix(2, 2, values=[[6, 5], [4, 3]]) flat_values = dm.as_sorted_flat_values([m1, m2]) self.assertEquals(4, len(flat_values)) self.assertTrue((flat_values[0] == [2, 3]).all()) self.assertTrue((flat_values[1] == [3, 4]).all()) self.assertTrue((flat_values[2] == [4, 5]).all()) self.assertTrue(np.isnan(flat_values[3][0])) self.assertEquals(6, flat_values[3][1])
def test_as_flat_values_empty_input(self): """tests as_sorted_flat_values() with no input""" self.assertTrue((dm.as_sorted_flat_values([]) == []).all())