def test_get_points_changes_state_depending_on_modulation(self, mock_listdir, mock_open): mock_listdir.return_value = ['BLA_1.0_.wav'] mock_wave = MagicMock() mock_open.return_value = mock_wave mock_wave.getparams.return_value = self.default_params(nframes=24) l = np.array( (-32768, 0, 32767, 0, -32768, 0, 32767, 0, -32768, -16384, 0, 16384, 32767, 16384, 0, -16384, 32768, -16384, 0, 16384, 32767, 16384, 0, -16384, ) ) data = np.column_stack((l, l)) wave_data = data.astype(np.dtype('<i2')).tostring() mock_wave.readframes.return_value = wave_data wfps = WavFolderPointSource(self.test_folder) expected_point_on = Point(1.0, 1.0, 1.0, True) expected_point_off = Point(1.0, 1.0, 1.0, False) points = list(wfps.get_points()) self.assertEquals(4, len(points)) self.assertEqual(expected_point_on, points[0]) self.assertEqual(expected_point_on, points[1]) self.assertEqual(expected_point_off, points[2]) self.assertEqual(expected_point_off, points[3])
def test_get_points_raises_exception_given_non_stereo_wav_file(self, mock_wave): wave_file = MagicMock() mock_wave.open.return_value = wave_file wave_file.getparams.return_value(self.default_params(nchannels=1)) wfps = WavFolderPointSource(self.test_folder) with self.assertRaises(Exception): points = wfps.get_points() points.next() mock_wave.open.assert_called_with(os.path.join(self.test_folder,'layer_69.88_.wav'), 'r')
def test_get_points_raises_exception_given_non_stereo_wav_file( self, mock_wave): wave_file = MagicMock() mock_wave.open.return_value = wave_file wave_file.getparams.return_value(self.default_params(nchannels=1)) wfps = WavFolderPointSource(self.test_folder) with self.assertRaises(Exception): points = wfps.get_points() points.next() mock_wave.open.assert_called_with( os.path.join(self.test_folder, 'layer_69.88_.wav'), 'r')
def test_get_points_returns_expected_point(self, mock_listdir, mock_open): mock_listdir.return_value = ['BLA_1.0_.wav'] mock_wave = MagicMock() mock_open.return_value = mock_wave mock_wave.getparams.return_value = self.default_params(nframes=4) l = np.array((-32768, 0, 32767, 0)) data = np.column_stack((l, l)) wave_data = data.astype(np.dtype('<i2')).tostring() mock_wave.readframes.return_value = wave_data wfps = WavFolderPointSource(self.test_folder) expected_point = Point(1.0, 1.0, 1.0, True) points = list(wfps.get_points()) self.assertEquals(1, len(points)) self.assertEqual(expected_point, points[0])
def test_get_points_changes_state_depending_on_modulation( self, mock_listdir, mock_open): mock_listdir.return_value = ['BLA_1.0_.wav'] mock_wave = MagicMock() mock_open.return_value = mock_wave mock_wave.getparams.return_value = self.default_params(nframes=24) l = np.array(( -32768, 0, 32767, 0, -32768, 0, 32767, 0, -32768, -16384, 0, 16384, 32767, 16384, 0, -16384, 32768, -16384, 0, 16384, 32767, 16384, 0, -16384, )) data = np.column_stack((l, l)) wave_data = data.astype(np.dtype('<i2')).tostring() mock_wave.readframes.return_value = wave_data wfps = WavFolderPointSource(self.test_folder) expected_point_on = Point(1.0, 1.0, 1.0, True) expected_point_off = Point(1.0, 1.0, 1.0, False) points = list(wfps.get_points()) self.assertEquals(4, len(points)) self.assertEqual(expected_point_on, points[0]) self.assertEqual(expected_point_on, points[1]) self.assertEqual(expected_point_off, points[2]) self.assertEqual(expected_point_off, points[3])
def displayListFromWaves(self, folder): wfps = WavFolderPointSource(folder) points = wfps.get_points() dlb = DisplayListBuilder() return dlb.get_list_id(points)
def test_init_raises_exception_if_valid_folder_but_no_wav_files(self): with self.assertRaises(Exception): WavFolderPointSource('.')
def test_init_returns_if_valid_folder(self): WavFolderPointSource(self.test_folder)
def test_init_raises_exception_if_folder_doesnt_exist(self): with self.assertRaises(Exception): WavFolderPointSource('ERRR')