class TestPlaylist(unittest.TestCase):
def setUp(self):
self.access_token = get_access_token()
self.track = Track("Blue in Green", "Miles Davis")
def test_query(self):
self.assertTrue(self.track.spotify_query(self.access_token))
def test_view_top_results(self):
self.assertEqual(self.track.view_top_results(self.access_token, limit=1), None)
def setUp(self):
# ------- Vertical track test ------- #
self.vertical_track_height = 200
# Set up track
line = Line.Line([[0, 0, self.vertical_track_height], [0, 0, 0]])
self.vertical_track = Track.Track([line])
# ------- Circular ------- #
self.radius = 1
# Set up track
vertical_circle = Circle.VerticalCircle(self.radius)
self.circular_track = Track.Track([vertical_circle])
def setUp(self):
# ------- Circular ------- #
self.radius = 1
# Set up track
vertical_circle = Circle.VerticalCircle(self.radius)
self.circular_track = Track.Track([vertical_circle])
def setUp(self):
# Three line track setup
self.line_1 = Line.Line([[0, 0, 0], [2, 0, 0]])
self.line_2 = Line.Line([[2, 0, 0], [2, 2, 0]])
self.line_3 = Line.Line([[2, 2, 0], [0, 2, 0]])
self.line_track = Track.Track([self.line_1, self.line_2, self.line_3])
# Two Bezier lines track setup
knots_1 = [[0, 0, 0], [12, 0, 0]]
control_points_1 = [[4, 0, 0], [8, 0, 0]]
self.bezier_1 = BezierSpline.CubicBezier(knots_1, control_points_1)
knots_2 = [[12, 0, 0], [12, 12, 0]]
control_points_2 = [[12, 4, 0], [12, 8, 0]]
self.bezier_2 = BezierSpline.CubicBezier(knots_2, control_points_2)
self.bezier_track = Track.Track([self.bezier_1, self.bezier_2])
def import_track(file_path, start_row=1, end_row=None, **segment_options):
"""
Create a Track instance from a coordinate CSV file
:return: Track instance
"""
return Track(
fit_cubic_bezier(
read_coordinate_file(file_path,
start_row=start_row,
end_row=end_row), **segment_options))
def _get_list_of_tracks(line, delimiter):
"""
Helper function for generating track lists from files and string blocks. Parses an individual track string into
a Track object
Parameters
----------
line: str
Contains artist and track separated by a delimiter
delimiter: str
Character(s) separating artist and track
Returns
-------
Track
"""
contents = line.strip().split(delimiter)
try:
return Track(contents[1].strip(), contents[0].strip())
except IndexError:
return Track(contents[0].strip(), contents[0].strip())
def setUp(self):
# ------- Vertical track test ------- #
self.vertical_track_height = 5
# Set up track
line_1 = Line.Line([[0, 0, self.vertical_track_height],
[0, 0, (3. / 4.) * self.vertical_track_height]])
line_2 = Line.Line([[0, 0, (3. / 4.) * self.vertical_track_height],
[0, 0, (2. / 4.) * self.vertical_track_height]])
line_3 = Line.Line([[0, 0, (2. / 4.) * self.vertical_track_height],
[0, 0, (1. / 4.) * self.vertical_track_height]])
line_4 = Line.Line([[0, 0, (1. / 4.) * self.vertical_track_height],
[0, 0, 0]])
self.vertical_track = Track.Track([line_1, line_2, line_3, line_4])
# ------- Circular ------- #
self.radius = 1
# Set up track
vertical_circle = Circle.VerticalCircle(self.radius)
self.circular_track = Track.Track([vertical_circle])
import pyglet
from pyglet import clock, shapes
from pyglet.window import key
from src.Track import Track
from src.Vector import Vector
from src.qlearning import QLearning
track = Track('../tracks/track05.png',
startPos=Vector(26, 3),
finishA=Vector(25, 1),
finishB=Vector(25, 5),
checkPointC=Vector(20, 44),
checkPointD=Vector(20, 48),
checkPointA=Vector(89, 8),
checkPointB=Vector(99, 8))
ql = QLearning(track, alpha=0.1, gamma=0.95)
# Window
tile_size = 10
window = pyglet.window.Window(width=track.width * tile_size,
height=(track.height * tile_size) + 200)
track_img = pyglet.image.load('../tracks/track05.png')
sprite = pyglet.sprite.Sprite(img=track_img, x=0, y=0)
sprite.scale = 10
f1_img = pyglet.image.load('f1.png')
sprite_f1 = pyglet.sprite.Sprite(img=f1_img, x=0, y=0)
sprite_f1.scale = 0.01
def setUp(self):
self.newTrack = Track("Born to Run", "Bruce Springsteen")
self.newSpotifyTrack = Track("Interplay", "Bill Evans", "fake/href",
"fake/externalURL", "jkwon")
def setUp(self):
self.access_token = get_access_token()
self.track = Track("Blue in Green", "Miles Davis")
def setUp(self):
# ------- Vertical track test ------- #
# Set up track
line = Line.Line([[0, 0, 0.5], [0, 0, 0]])
self.vertical_track = Track.Track([line])
def test_set_tracks(self):
self.newPlaylist.track = Track("EARFQUAKE", "Tyler, The Creator")
self.assertEqual(self.newPlaylist.track.song, "EARFQUAKE")
self.assertEqual(self.newPlaylist.track.artist, "Tyler, The Creator")
import pyglet
from pyglet import clock, shapes
from pyglet.window import key
from src.Track import Track
from src.Vector import Vector
from src.qlearning import QLearning
track_img = 'tracks/track05.png'
track = Track(track_img,
startPos=Vector(26, 3),
finishA=Vector(25, 1),
finishB=Vector(25, 5),
checkPointC=Vector(20, 44),
checkPointD=Vector(20, 48),
checkPointA=Vector(89, 8),
checkPointB=Vector(99, 8))
ql = QLearning(track, alpha=0.1, gamma=0.95)
# Window
tile_size = 10
window = pyglet.window.Window(width=track.width * tile_size,
height=(track.height * tile_size) + 200)
track_img = pyglet.image.load(track_img)
sprite = pyglet.sprite.Sprite(img=track_img, x=0, y=0)
sprite.scale = 10
f1_img = pyglet.image.load('ui/f1.png')