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())
