def test_reverse_map(self): # sort_order ascending myarray = arange(10) data_source = ArrayDataSource(myarray, sort_order='ascending') self.assertEqual(data_source.reverse_map(4.0), 4) # sort_order descending myarray = arange(10)[::-1] data_source = ArrayDataSource(myarray, sort_order='descending') self.assertEqual(data_source.reverse_map(4.0), 5) # sort_order none myarray = array([12, 3, 0, 9, 2, 18, 3]) data_source = ArrayDataSource(myarray, sort_order='none') with self.assertRaises(NotImplementedError): data_source.reverse_map(3)